Searched defs:add_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 259 auto add_shape = ctx->input(3).shape(); in ExtendMklMatMulParams() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 1023 auto add_shape = ops::Placeholder::Shape({6}); in TEST_F() local
1088 auto add_shape = ops::Placeholder::Shape({1, 1, 1, 1, 6}); in TEST_F() local
1154 auto add_shape = ops::Placeholder::Shape({1, 1, 6}); in TEST_F() local
1915 auto add_shape = ops::Placeholder::Shape({8, 4, 32, 32, 128}); in TEST_F() local
1987 auto add_shape = ops::Placeholder::Shape({8, 4, 32, 32, 128}); in TEST_F() local
2059 auto add_shape = ops::Placeholder::Shape({8, 4, 32, 32, 128}); in TEST_F() local
|
| D | mkl_remapper_test.cc | 759 auto add_shape = TensorShape({b0, 1, m, n}); in VerifyFused() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | serialization_base_generated.h | 1211 void add_shape(flatbuffers::Offset<tflite::gpu::data::BHWDC> shape) { in add_shape() function
|