HomeSort by relevance Sort by last modified time
    Searched refs:shapes_and_types (Results 1 - 8 of 8) sorted by null

  /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 192 const auto* shapes_and_types = c.output_handle_shapes_and_types(0); local
193 ASSERT_TRUE(shapes_and_types != nullptr);
194 ASSERT_EQ(1, shapes_and_types->size());
195 EXPECT_EQ((*shapes_and_types)[0].dtype, DT_BOOL);
    [all...]
  /external/tensorflow/tensorflow/core/framework/
shape_inference.cc     [all...]
shape_inference.h 558 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
563 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
576 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
581 const std::vector<ShapeAndType>& shapes_and_types) TF_MUST_USE_RESULT;
596 int idx, const std::vector<ShapeAndType>& shapes_and_types) {
598 new std::vector<ShapeAndType>(shapes_and_types));
696 const std::vector<ShapeAndType>& shapes_and_types,
701 const std::vector<ShapeAndType>& shapes_and_types,
shape_inference_test.cc     [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
graph_properties.h 75 // Merges shapes <shapes_and_types>, determined from an EnqueueV2 node, into
79 const std::vector<shape_inference::ShapeAndType>& shapes_and_types,
81 // Relaxes shapes <shapes_and_types>, determined from an EnqueueV2 node, into
85 const std::vector<shape_inference::ShapeAndType>& shapes_and_types,
graph_properties.cc 641 const std::vector<ShapeAndType>& shapes_and_types,
643 if (shapes_and_types.size() != queue_shapes_and_types->size()) {
645 "Enqueue nodes mixed number of tensors: ", shapes_and_types.size(),
648 for (size_t i = 0; i < shapes_and_types.size(); ++i) {
649 const ShapeAndType& a = shapes_and_types[i];
664 const std::vector<ShapeAndType>& shapes_and_types,
666 if (shapes_and_types.size() != queue_shapes_and_types->size()) {
668 "Enqueue nodes mixed number of tensors: ", shapes_and_types.size(),
671 for (size_t i = 0; i < shapes_and_types.size(); ++i) {
672 const ShapeAndType& a = shapes_and_types[i]
893 std::vector<ShapeAndType> shapes_and_types; local
    [all...]
  /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) {

Completed in 100 milliseconds