Searched refs:ToTensorShape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | partial_tensor_shape_test.cc | 121 TEST(PartialTensorShapeTest, ToTensorShape) { in TEST() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 260 tensorflow::TensorShape ToTensorShape(llvm::ArrayRef<T> sizes) { in ToTensorShape() function
266 tensorflow::TensorShape ToTensorShape( in ToTensorShape() function
4157 ToTensorShape<int32_t, num_dims>(input_shape), in matchAndRewrite()
4158 ToTensorShape<int64_t, num_dims>(filter_shape), in matchAndRewrite()
4159 ToTensorShape<int64_t, num_dims>(out_backprop_ty.getShape()), in matchAndRewrite()
4340 ToTensorShape<int64_t, num_dims>(input_shape), in matchAndRewrite()
4341 ToTensorShape<int32_t, num_dims>(filter_shape), in matchAndRewrite()
4342 ToTensorShape<int64_t, num_dims>(out_backprop_shape), dilations, in matchAndRewrite()
|