Searched defs:shape_b (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass_test.cc | 305 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
357 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
382 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 191 Shape shape_b = ShapeUtil::MakeShape(F32, {4, 3}, {true, false}); in TEST() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1579 const auto shape_b = properties.GetOutputProperties("b").at(0).shape(); in TEST_F() local
1651 const auto shape_b = properties.GetOutputProperties("b").at(0).shape(); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 90 const RuntimeShape* shape_b = in ProcessBroadcastShapes() local
|