Searched refs:dummy_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 43 static const TensorShape dummy_shape({8}); variable
214 AddInputFromArray<uint8>(dummy_shape, dummy_tensor); in RunConv2D()
320 AddInputFromArray<uint8>(dummy_shape, dummy_tensor); in VerifyFusedBatchNorm()
451 AddInputFromArray<uint8>(dummy_shape, dummy_tensor); in VerifyFusedBatchNormGradWithConv2D()
454 AddInputFromArray<uint8>(dummy_shape, dummy_tensor); in VerifyFusedBatchNormGradWithConv2D()
455 AddInputFromArray<uint8>(dummy_shape, dummy_tensor); in VerifyFusedBatchNormGradWithConv2D()
456 AddInputFromArray<uint8>(dummy_shape, dummy_tensor); in VerifyFusedBatchNormGradWithConv2D()
457 AddInputFromArray<uint8>(dummy_shape, dummy_tensor); in VerifyFusedBatchNormGradWithConv2D()
|
| D | mkl_fused_ops_test.cc | 43 static const TensorShape dummy_shape({8}); variable
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_multistream_test.cc | 64 Shape dummy_shape = ShapeUtil::MakeShape(S32, {dummy_size}); in TEST() local
76 dummy_inputs.data(), S32, dummy_shape.dimensions(), in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 2081 TensorShape dummy_shape({8}); in GetDummyMklTensorNode() local
2082 dummy_shape.AsProto(proto.mutable_tensor_shape()); in GetDummyMklTensorNode()
|