Home
last modified time | relevance | path

Searched refs:ToShape (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dshape.h466 return this->ToShape() == shape.ToShape();
471 return this->ToShape() != shape.ToShape();
476 Shape ToShape() const {
583 return ToString(s.ToShape());
Dshape_test.cc115 Shape s2 = oihw.ToShape(); in TEST()
/external/tensorflow/tensorflow/core/util/
Dbcast.cc30 TensorShape BCast::ToShape(const BCastList::Vec& vec) { in ToShape() function in tensorflow::BCast
Dbcast.h417 static TensorShape ToShape(const Vec& vec);
/external/tensorflow/tensorflow/core/kernels/
Dcwise_ops_common.cc77 const TensorShape output_shape = BCast::ToShape(bcast.output_shape()); in BinaryOpState()
Dbroadcast_to_op.cc85 OP_REQUIRES(ctx, BCast::ToShape(bcast.output_shape()) == output_shape, in Compute()
Dxent_op.cc54 shape_in = BCast::ToShape(bcast.output_shape()); in Compute()
Drandom_binomial_op.cc362 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute()
479 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute()
Dsubstr_op.cc141 TensorShape output_shape = BCast::ToShape(bcast.result_shape()); in Compute()
Dcwise_op_select.cc186 const TensorShape output_shape = BCast::ToShape(bcast.output_shape()); in Compute()
Dquantized_mul_op.cc301 0, BCast::ToShape(bcast.output_shape()), &z)); in Compute()
Dquantized_add_op.cc474 0, BCast::ToShape(bcast.output_shape()), &z)); in Compute()
Dparameterized_truncated_normal_op.cc769 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dselect_op.cc109 BCast::ToShape(bcast_then_else.output_shape()).DebugString(), " vs. ", in Compile()