Searched defs:input_shapes (Results 1 - 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
H A Dbatch_dataset_op.cc54 const auto& input_shapes = input_->output_shapes(); local
55 output_shapes_.reserve(input_shapes.size());
56 for (const auto& input_shape : input_shapes) {
H A Dpadded_batch_dataset_op.cc112 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) {
H A Dtensor_queue_dataset_op.cc172 const auto& input_shapes = dataset()->input_->output_shapes(); variable
216 shapes[i].IsIdenticalTo(input_shapes[i])) {
/external/tensorflow/tensorflow/core/profiler/internal/
H A Dtfprof_show_multi.cc171 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) {
H A Dtfprof_node.h463 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/contrib/lite/toco/
H A Dmodel_cmdline_flags.cc66 "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
[all...]
H A Dargs.h192 Arg<string> input_shapes; member in struct:toco::ParsedModelFlags
/external/tensorflow/tensorflow/python/framework/
H A Dcpp_shape_inference.cc75 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,
/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dtest_util.cc156 std::vector<std::vector<int>> input_shapes) {
187 for (const auto& shape : input_shapes) {
155 BuildInterpreter( std::vector<std::vector<int>> input_shapes) argument
H A Doptional_tensor_test.cc36 const std::vector<std::vector<int>>& input_shapes)
109 BuildInterpreter(input_shapes);
33 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, bool use_peephole, bool use_projection_weights, bool use_projection_bias, float cell_clip, float proj_clip, const std::vector<std::vector<int>>& input_shapes) argument
H A Dlstm_test.cc38 const std::vector<std::vector<int>>& input_shapes)
111 BuildInterpreter(input_shapes);
35 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, bool use_peephole, bool use_projection_weights, bool use_projection_bias, float cell_clip, float proj_clip, const std::vector<std::vector<int>>& input_shapes) argument
H A Dunidirectional_sequence_lstm_test.cc40 const std::vector<std::vector<int>>& input_shapes)
115 BuildInterpreter(input_shapes);
35 UnidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, int sequence_length, bool use_cifg, bool use_peephole, bool use_projection_weights, bool use_projection_bias, float cell_clip, float proj_clip, const std::vector<std::vector<int>>& input_shapes) argument
/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compiler.cc431 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));
427 BuildArguments( const Graph& graph, const std::vector<XlaCompiler::Argument>& args, bool use_tuple_arg, xla::ComputationBuilder* builder, XlaContext* context, std::vector<int>* arg_cores, std::vector<XlaExpression>* arg_expressions, std::vector<int>* input_mapping, std::vector<xla::Shape>* input_shapes, bool is_entry_computation) argument
/external/tensorflow/tensorflow/core/common_runtime/
H A Dconstant_folding.cc54 // 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,
56 ReadPartialShapesFromShapeMap( const Node* n, const std::unordered_map<string, std::vector<PartialTensorShape>>* shape_map, std::vector<PartialTensorShape>* input_shapes) argument
79 MaybeReplaceShapeOrShapeNOp( const Node* n, const std::vector<PartialTensorShape>& input_shapes, std::unordered_map<const Node*, std::vector<Tensor>>* shape_replacement_map) argument
120 MaybeReplaceRankOp(const Node* n, const std::vector<PartialTensorShape>& input_shapes, std::unordered_map<const Node*, std::vector<Tensor>>* shape_replacement_map) argument
136 MaybeReplaceSizeOp(const Node* n, const std::vector<PartialTensorShape>& input_shapes, std::unordered_map<const Node*, std::vector<Tensor>>* shape_replacement_map) argument
181 std::vector<PartialTensorShape> input_shapes; local
[all...]
/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_concat_op.cc155 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);
154 Compute(OpKernelContext* c, const std::vector<Tensor>& values, const TensorShapeList& input_shapes) argument
433 MklCreateInputLayouts(OpKernelContext* context, MklShapeList& input_shapes) argument
489 CallEigenVersion(OpKernelContext* context, const OpInputList& values, const MklShapeList& input_shapes) argument
531 CallEigenVersion(OpKernelContext* context, const OpInputList& values, const TensorShapeList& input_shapes) argument
784 CallEigenVersion(OpKernelContext* context, const OpInputList& values, const TensorShapeList& input_shapes) argument
[all...]
/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.cc34 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
1027 input_shapes
32 InferenceContext( int graph_def_version, const NodeDef* node_def, const OpDef& op_def, const std::vector<TensorShapeProto>& input_shapes, const std::vector<const Tensor*>& input_tensors, const std::vector<TensorShapeProto>& input_tensors_as_shapes, const std::vector< std::unique_ptr<std::vector<std::pair<TensorShapeProto, DataType>>>>& input_handle_shapes_and_types) argument
85 InferenceContext( int graph_def_version, const NodeDef* node_def, const OpDef& op_def, const std::vector<PartialTensorShape>& input_shapes, const std::vector<const Tensor*>& input_tensors, const std::vector<PartialTensorShape>& input_tensors_as_shapes, const std::vector< std::unique_ptr<std::vector<std::pair<PartialTensorShape, DataType>>>>& input_handle_shapes_and_types) argument
136 InferenceContext( int graph_def_version, const NodeDef* node_def, const OpDef& op_def, const std::vector<ShapeHandle>& input_shapes, const std::vector<const Tensor*>& input_tensors, const std::vector<ShapeHandle>& input_tensors_as_shapes, std::vector<std::unique_ptr<std::vector<ShapeAndType>>> input_handle_shapes_and_types) argument
[all...]

Completed in 1990 milliseconds