Searched defs:CheckShape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ | ||
| D | hlo_verifier.cc | 846 Status ShapeVerifier::CheckShape(const HloInstruction* instruction, in CheckShape() function in xla::ShapeVerifier |
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ | ||
| D | fused_conv2d_bias_activation_op.cc | 154 Status CheckShape(const Tensor& tensor, const string& tensor_name) { in CheckShape() function in tensorflow::FusedConv2DBiasActivationOp |
| /external/tensorflow/tensorflow/compiler/xla/tests/ | ||
| D | batch_normalization_test.cc | 73 XlaOp CheckShape(XlaBuilder* b, const XlaOp& operand, in CheckShape() function in xla::__anon28118db90111::BatchNormalizationTest |