Searched refs:input_tensors_as_shapes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 256 std::vector<ShapeHandle> input_tensors_as_shapes; in AddNode() local
261 input_shapes, input_tensors, input_tensors_as_shapes, in AddNode()
629 std::vector<ShapeHandle> input_tensors_as_shapes; in RunShapeFn() local
634 c->set_input_tensors_as_shapes(input_tensors_as_shapes); in RunShapeFn()
702 if (i >= input_tensors_as_shapes.size()) { in RunShapeFn()
703 input_tensors_as_shapes.resize(i + 1); in RunShapeFn()
707 input_tensors_as_shapes[i] = s; in RunShapeFn()
716 c->set_input_tensors_as_shapes(input_tensors_as_shapes); in RunShapeFn()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 177 const std::vector<ShapeHandle>& input_tensors_as_shapes,
196 const std::vector<TensorShapeProto>& input_tensors_as_shapes,
216 const std::vector<PartialTensorShape>& input_tensors_as_shapes,
318 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in set_input_tensors_as_shapes() argument
319 input_tensors_as_shapes_ = input_tensors_as_shapes; in set_input_tensors_as_shapes()
322 const std::vector<ShapeHandle>& input_tensors_as_shapes() const { in input_tensors_as_shapes() function
693 const std::vector<ShapeHandle>& input_tensors_as_shapes);
|
| D | shape_inference.cc | 36 const std::vector<TensorShapeProto>& input_tensors_as_shapes, in InferenceContext() argument
43 input_tensors_as_shape_handles.reserve(input_tensors_as_shapes.size()); in InferenceContext()
44 for (const TensorShapeProto& p : input_tensors_as_shapes) { in InferenceContext()
91 const std::vector<PartialTensorShape>& input_tensors_as_shapes, in InferenceContext() argument
98 input_tensors_as_shape_handles.reserve(input_tensors_as_shapes.size()); in InferenceContext()
99 for (const PartialTensorShape& p : input_tensors_as_shapes) { in InferenceContext()
144 const std::vector<ShapeHandle>& input_tensors_as_shapes, in InferenceContext() argument
149 PreInputInit(op_def, input_tensors, input_tensors_as_shapes); in InferenceContext()
226 const std::vector<ShapeHandle>& input_tensors_as_shapes) { in PreInputInit() argument
228 input_tensors_as_shapes_ = input_tensors_as_shapes; in PreInputInit()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 673 } else if (ic->input_tensors_as_shapes().size() > i && in UpdateFunction()
675 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in UpdateFunction()
680 ic, ic->input(i), ic->input_tensors_as_shapes()[i], in UpdateFunction()
758 std::vector<ShapeHandle> input_tensors_as_shapes(ic->num_inputs()); in UpdateNode() local
786 input_tensors_as_shapes[dst_input] = in UpdateNode()
797 if (!ic->FullyDefined(input_tensors_as_shapes[dst_input])) { in UpdateNode()
823 &input_tensors_as_shapes[dst_input]); in UpdateNode()
865 ic->set_input_tensors_as_shapes(input_tensors_as_shapes); in UpdateNode()
1088 std::vector<ShapeHandle> input_tensors_as_shapes; in AddNode() local
1092 input_tensors, input_tensors_as_shapes, in AddNode()
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes.py | 691 input_tensors_as_shapes = arr
698 input_tensors_as_shapes, status)
|