Searched defs:shape_a (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | process_broadcast_shapes.h | 86 const RuntimeShape* shape_a = in ProcessBroadcastShapes() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass_test.cc | 300 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
352 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
377 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_op_test.cc | 147 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_a = in BuildRemoteFusedGraphExecuteInfo() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1586 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
1670 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
1776 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 211 Shape shape_a = ShapeUtil::MakeShape(F32, {4, 3}, {true, false}); in TEST() local
|