| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_concat_op.cc | 155 const TensorShapeList& input_shapes) {
171 const int input_dims = input_shapes[0].dims();
172 const TensorShape& input_shape = input_shapes[0];
197 const bool in_is_scalar = IsLegacyScalar(input_shapes[i]);
200 (input_shapes[i].dims() == input_dims) ||
205 "] = ", input_shapes[i].DebugString()));
212 input_shapes[i].dims() > 0 ? input_shapes[i].dim_size(axis) : 1;
260 MklShapeList input_shapes(N);
261 GetMklShapeList(context, "values", &input_shapes);
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| constant_folding.cc | 54 // stores it to input_shapes. Returns false if any input does not have a shape
60 std::vector<PartialTensorShape>* input_shapes) {
72 input_shapes->push_back(known_shape[in->src_output()]);
80 const Node* n, const std::vector<PartialTensorShape>& input_shapes,
84 for (const auto& shape : input_shapes) {
121 const std::vector<PartialTensorShape>& input_shapes,
124 CHECK_EQ(input_shapes.size(), 1);
125 if (input_shapes[0].unknown_rank()) {
129 t.scalar<int32>()() = input_shapes[0].dims();
137 const std::vector<PartialTensorShape>& input_shapes,
181 std::vector<PartialTensorShape> input_shapes; local
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| model_cmdline_flags.cc | 66 "Deprecated: use --input_shapes instead. Input array shape. For "
70 Flag("input_shapes", parsed_flags.input_shapes.bind(),
71 parsed_flags.input_shapes.default_value(),
211 parsed_model_flags.input_shapes.specified();
216 "flags (--input_arrays, --input_shapes, --mean_values, --std_values), "
294 if (parsed_model_flags.input_shapes.specified()) {
296 std::vector<string> input_shapes = local
297 absl::StrSplit(parsed_model_flags.input_shapes.value(), ':');
298 QCHECK(input_shapes.size() == model_flags->input_arrays_size())
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| conv2d_benchmark.py | 147 input_shapes = []
151 input_shapes += [[b, h, w, c]]
155 for ishape, fshape in zip(input_shapes, filter_shapes):
|
| special_math_ops_test.py | 256 for axes, input_shapes in self.dim_mismatch_cases:
259 for shape in input_shapes
|
| /external/tensorflow/tensorflow/python/framework/ |
| cpp_shape_inference.cc | 75 std::vector<TensorShapeProto> input_shapes; local
79 input_shapes.resize(input_serialized_shapes.size());
89 input_shapes[i].Swap(tmp.mutable_shape());
128 graph_def_version, &node, op_reg_data->op_def, input_shapes,
|
| common_shapes.py | 664 input_shapes = [tensor_to_inference_result(i) for i in op.inputs]
666 input_tensors = [None for i in input_shapes]
674 arr = [serialized_unknown_shape for i in input_shapes]
685 graph_def_version, node_def_str, input_shapes, input_tensors,
|
| /external/tensorflow/tensorflow/python/layers/ |
| network.py | 796 input_shapes = []
799 input_shapes.append(tuple(tensor_shape.TensorShape(shape).as_list()))
801 input_shapes.append(None)
804 input_shapes = [tuple(tensor_shape.TensorShape(input_shape).as_list())]
806 input_shapes = [None]
808 if len(input_shapes) != len(self._input_layers):
813 cache_key = layers_util.object_list_uid(input_shapes)
818 for i in range(len(input_shapes)):
820 input_shape = input_shapes[i]
841 input_shapes = [
[all...] |
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| tfprof_show_multi.cc | 171 std::map<int, std::vector<int64>> input_shapes; local
172 for (const auto& inp : gnode.input_shapes()) {
173 input_shapes[inp.first] = ShapeProtoToVec(inp.second);
177 for (const auto& s : input_shapes) {
|
| tfprof_node.h | 463 for (const auto& s : node.input_shapes()) {
705 const std::map<int, std::vector<int64>> input_shapes() const { function in class:tensorflow::tfprof::TFGraphNode
706 std::map<int, std::vector<int64>> input_shapes; local
709 std::vector<int64>& shape_vec = input_shapes[inp.first];
727 return input_shapes;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| batch_dataset_op.cc | 54 const auto& input_shapes = input_->output_shapes(); local
55 output_shapes_.reserve(input_shapes.size());
56 for (const auto& input_shape : input_shapes) {
|
| padded_batch_dataset_op.cc | 112 const auto& input_shapes = input_->output_shapes(); local
113 output_shapes_.reserve(input_shapes.size());
114 for (size_t i = 0; i < input_shapes.size(); ++i) {
|
| /external/tensorflow/tensorflow/core/framework/ |
| shape_inference.cc | 34 const std::vector<TensorShapeProto>& input_shapes,
53 for (const TensorShapeProto& p : input_shapes) {
63 input_shapes.size());
87 const std::vector<PartialTensorShape>& input_shapes,
106 for (const PartialTensorShape& p : input_shapes) {
115 input_shapes.size());
138 const std::vector<ShapeHandle>& input_shapes,
147 inputs_ = input_shapes;
1025 std::vector<string> input_shapes; local
[all...] |
| shape_inference.h | 163 // <input_tensors> is NULL-padded to be the same size as <input_shapes>.
169 // <input_tensors_as_shapes> can have fewer elements than <input_shapes>.
175 const std::vector<ShapeHandle>& input_shapes,
181 // <input_tensors> is NULL-padded to be the same size as <input_shapes>.
187 // can have fewer elements than <input_shapes>. Values of
194 const std::vector<TensorShapeProto>& input_shapes,
201 // <input_tensors> is NULL-padded to be the same size as <input_shapes>.
207 // can have fewer elements than <input_shapes>. Values of
214 const std::vector<PartialTensorShape>& input_shapes,
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| optional_tensor_test.cc | 36 const std::vector<std::vector<int>>& input_shapes)
109 BuildInterpreter(input_shapes);
|
| test_util.cc | 156 std::vector<std::vector<int>> input_shapes) {
187 for (const auto& shape : input_shapes) {
|
| lstm_test.cc | 38 const std::vector<std::vector<int>>& input_shapes)
111 BuildInterpreter(input_shapes);
[all...] |
| unidirectional_sequence_lstm_test.cc | 40 const std::vector<std::vector<int>>& input_shapes)
115 BuildInterpreter(input_shapes);
[all...] |
| test_util.h | 150 void BuildInterpreter(std::vector<std::vector<int>> input_shapes);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_compiler.h | 319 std::vector<xla::Shape>* input_shapes,
|
| xla_compiler.cc | 431 std::vector<int>* input_mapping, std::vector<xla::Shape>* input_shapes,
492 input_shapes->push_back(xla::ShapeUtil::MakeTupleShape(arg_shapes));
494 *input_shapes = arg_shapes;
533 tuple = builder->Parameter(0, (*input_shapes)[0], "arg_tuple");
535 tuple = builder->Parameter(0, (*input_shapes)[0], "arg_tuple");
551 builder->Parameter(i, (*input_shapes)[i], strings::StrCat("arg", i));
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/utils/ |
| vis_utils.py | 109 elif hasattr(layer, 'input_shapes'):
110 inputlabels = ', '.join([str(ishape) for ishape in layer.input_shapes])
|
| /frameworks/ml/nn/common/operations/ |
| LSTMTest.cpp | 89 std::vector<std::vector<uint32_t>> input_shapes(input_shapes0);
91 input_shapes.push_back({n_batch, n_output});
92 input_shapes.push_back({n_batch, n_cell});
93 auto it = input_shapes.begin();
[all...] |
| SVDFTest.cpp | 193 std::vector<std::vector<uint32_t>> input_shapes{
201 auto it = input_shapes.begin();
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/layers/ |
| merge.py | 393 input_shapes = input_shape
394 output_shape = list(input_shapes[0])
395 for shape in input_shapes[1:]:
|
