Home
last modified time | relevance | path

Searched refs:IsIdenticalTo (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_tensor_test.cc48 EXPECT_TRUE(output_shape.IsIdenticalTo(PartialTensorShape())); in TEST()
58 EXPECT_TRUE(output_shape.IsIdenticalTo(PartialTensorShape({-1, 2, 3}))); in TEST()
68 EXPECT_TRUE(output_shape.IsIdenticalTo(PartialTensorShape({1, 2, 3}))); in TEST()
76 EXPECT_TRUE(output_shape.IsIdenticalTo(TensorShape())); in TEST()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dsignature_flattening_test.cc35 PartialTensorShape(spec.shape()).IsIdenticalTo(expected_shape) && in TensorSpecsAreEqual()
/external/tensorflow/tensorflow/core/framework/
Dpartial_tensor_shape_test.cc149 EXPECT_TRUE(shapes[i].IsIdenticalTo(shapes[j])); in TEST()
151 EXPECT_FALSE(shapes[i].IsIdenticalTo(shapes[j])); in TEST()
Dtensor_shape.cc935 bool PartialTensorShape::IsIdenticalTo(const PartialTensorShape& shape) const { in IsIdenticalTo() function in tensorflow::PartialTensorShape
990 if (!shapes0[i].IsIdenticalTo(shapes1[i])) { in AreIdentical()
Dtensor_shape.h541 bool IsIdenticalTo(const PartialTensorShape& shape) const;
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_expression_test.cc81 EXPECT_TRUE(op_.IsIdenticalTo(op_as_op)); in TEST_F()
Dxla_compiler.cc288 bool modified = !resource->value().IsIdenticalTo(resource->initial_value()); in BuildComputation()
294 !grad.second->value().IsIdenticalTo(grad.second->initial_value()) || in BuildComputation()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib.cc454 if (!first_shape.IsIdenticalTo(component_shape)) { in Shape()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle_test.cc190 EXPECT_EQ(dtypes_and_shapes.at(0).shape.IsIdenticalTo({2, 2}), true); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.cc521 OP_REQUIRES(c, l_a->element_shape.IsIdenticalTo(l_b->element_shape), in Compute()
/external/tensorflow/tensorflow/cc/framework/
Dcc_op_gen.cc142 if (shape.IsIdenticalTo(PartialTensorShape())) { in PrintTensorShape()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h100 bool IsIdenticalTo(XlaOp rhs) const { in IsIdenticalTo() function