Searched refs:maybe_cast (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/ |
| H A D | unified_api_test.py | 50 def maybe_cast(t, perform_cast): function
421 x = maybe_cast(array_ops.ones([batch_size, image_size]), cast)
425 maybe_cast(
428 softmax_weight = maybe_cast(
430 labels = maybe_cast(array_ops.zeros([batch_size], dtype=dtypes.int32), cast)
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | common.py | 2288 def maybe_cast(self, model, example_inputs): member in BenchmarkRunner
2497 model, example_inputs = self.maybe_cast(model, example_inputs)
2666 model, example_inputs = self.maybe_cast(model, example_inputs)
2733 return experiment(*self.maybe_cast(model, example_inputs))
2761 model, example_inputs = self.maybe_cast(model, example_inputs)
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 2309 auto maybe_cast = [&](Value val, Type t) -> Value { in matchAndRewrite() local
2313 last_val[0] = maybe_cast(const_val, op.getResult(3).getType()); in matchAndRewrite()
2314 last_val[1] = maybe_cast(const_val, op.getResult(4).getType()); in matchAndRewrite()
|