| /external/tensorflow/tensorflow/core/ops/ |
| data_flow_ops_test.cc | 205 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types; local
206 op.input_resource_handle_shapes_and_types.push_back(&shapes_and_types);
210 shapes_and_types.emplace_back("[1,?,3]", DT_FLOAT);
214 shapes_and_types.emplace_back("[?,2]", DT_FLOAT);
229 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types; local
230 op.input_resource_handle_shapes_and_types.push_back(&shapes_and_types);
233 shapes_and_types.emplace_back("[1,?,3]", DT_FLOAT);
236 shapes_and_types.emplace_back("[?,2]", DT_FLOAT);
245 shapes_and_types.clear();
248 op.input_resource_handle_shapes_and_types.push_back(&shapes_and_types);
279 std::vector<ShapeInferenceTestOp::ShapeAndType> shapes_and_types; local
[all...] |
| array_ops_test.cc | 207 const auto* shapes_and_types = c.output_handle_shapes_and_types(0); local
208 ASSERT_TRUE(shapes_and_types != nullptr);
209 ASSERT_EQ(1, shapes_and_types->size());
210 EXPECT_EQ((*shapes_and_types)[0].dtype, DT_BOOL);
[all...] |
| /external/tensorflow/tensorflow/c/ |
| python_api.cc | 132 const auto* shapes_and_types = local
134 if (shapes_and_types == nullptr) return "";
136 for (const auto& p : *shapes_and_types) {
161 std::vector<tensorflow::shape_inference::ShapeAndType> shapes_and_types; local
167 shapes_and_types.emplace_back(shape, shape_and_type_proto.dtype());
169 ic->set_output_handle_shapes_and_types(output.index, shapes_and_types);
|
| /external/tensorflow/tensorflow/python/framework/ |
| cpp_shape_inference.cc | 142 const auto* shapes_and_types = c.output_handle_shapes_and_types(i); local
143 if (shapes_and_types != nullptr) {
146 for (const auto& p : *shapes_and_types) {
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties.cc | 1971 std::vector<ShapeAndType> shapes_and_types; local
2014 std::vector<ShapeAndType> shapes_and_types; local
[all...] |
