Home
last modified time | relevance | path

Searched defs:shape_a (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass_test.cc303 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
355 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
380 Output shape_a = ops::Placeholder(s.WithOpName("shape_a"), DT_INT32, in TEST() local
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test.cc147 RemoteFusedGraphExecuteInfo::TensorShapeTypeProto& shape_a = in BuildRemoteFusedGraphExecuteInfo() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc1566 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
1650 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
1756 const auto shape_a = properties.GetOutputProperties("a").at(0).shape(); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc189 Shape shape_a = ShapeUtil::MakeShape(F32, {4, 3}, {true, false}); in TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h88 const RuntimeShape* shape_a = in ProcessBroadcastShapes() local