Searched defs:shape_b (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_input_fusion.cc | 63 const Shape& shape_b) { in CompareShapeDimsFromLeftToRight()
98 Shape shape_b = GetInputShapeForMultiOutputFusion(*b); in FindAndSortFusionCandidates() local
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | resource_variable_test.cc | 99 std::vector<int> shape_b = {1}; in TEST() local
140 std::vector<int> shape_b = {2}; in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | process_broadcast_shapes.h | 92 const RuntimeShape* shape_b = in ProcessBroadcastShapes() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass_test.cc | 302 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
354 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
379 Output shape_b = ops::Placeholder(s.WithOpName("shape_b"), DT_INT32, in TEST() local
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | saved_model_api_test.cc | 236 const TF_Shape* shape_b = TF_TensorSpecShape(tensor_spec_b); in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 215 Shape shape_b = ShapeUtil::MakeShape(F32, {4, 3}, {true, false}); in TEST() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1821 const auto shape_b = properties.GetOutputProperties("b").at(0).shape(); in TEST_F() local
1893 const auto shape_b = properties.GetOutputProperties("b").at(0).shape(); in TEST_F() local
|