Searched defs:shape_a (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_input_fusion.cc | 62 bool CompareShapeDimsFromLeftToRight(const Shape& shape_a, in CompareShapeDimsFromLeftToRight()
97 Shape shape_a = GetInputShapeForMultiOutputFusion(*a); in FindAndSortFusionCandidates() local
|
| /external/tensorflow/tensorflow/lite/experimental/resource/ |
| D | resource_variable_test.cc | 98 std::vector<int> shape_a = {1}; in TEST() local
139 std::vector<int> shape_a = {1}; in TEST() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | process_broadcast_shapes.h | 90 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/c/experimental/saved_model/internal/ |
| D | saved_model_api_test.cc | 227 const TF_Shape* shape_a = TF_TensorSpecShape(tensor_spec_a); in TEST_P() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 1808 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
1892 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
2000 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 213 Shape shape_a = ShapeUtil::MakeShape(F32, {4, 3}, {true, false}); in TEST() local
|