Searched refs:CheckShape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 125 return CheckShape(concatenate, in HandleConcatenate()
131 return CheckShape(convert, ShapeInference::InferConvertShape( in HandleConvert()
137 return CheckShape(convert, ShapeInference::InferBitcastConvertShape( in HandleBitcastConvert()
151 return CheckShape(dot, expected); in HandleDot()
161 return CheckShape(convolution, expected); in HandleConvolution()
169 return CheckShape(fft, expected); in HandleFft()
177 return CheckShape(hlo, expected); in HandleTriangularSolve()
184 return CheckShape(hlo, expected); in HandleCholesky()
192 return CheckShape(crs, ShapeInference::InferAllReduceShape(operand_shapes)); in HandleAllReduce()
200 return CheckShape(hlo, in HandleAllToAll()
[all …]
|
| D | hlo_verifier.h | 108 Status CheckShape(const HloInstruction* instruction,
112 Status CheckShape(const HloInstruction* instruction,
|
| /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
175 OP_REQUIRES_OK(context, CheckShape(conv_input, "conv_input")); in Compute()
180 OP_REQUIRES_OK(context, CheckShape(filter, "filter")); in Compute()
184 OP_REQUIRES_OK(context, CheckShape(bias, "bias")); in Compute()
198 OP_REQUIRES_OK(context, CheckShape(side_input, "side_input")); in Compute()
|
| /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
175 CheckShape(&builder, ConstantLiteral(&builder, input_literal_), in XLA_TEST_P()
182 auto sum = CheckShape( in XLA_TEST_P()
201 CheckShape(&builder, in XLA_TEST_P()
208 CheckShape(&builder, Gt(standard_deviation, epsilon), in XLA_TEST_P()
|