/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | 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) {
|
H A D | 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) {
|
H A D | tensor_queue_dataset_op.cc | 172 const auto& input_shapes = dataset()->input_->output_shapes(); variable 216 shapes[i].IsIdenticalTo(input_shapes[i])) {
|
/external/tensorflow/tensorflow/core/profiler/internal/ |
H A D | 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) {
|
H A D | 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/contrib/lite/toco/ |
H A D | 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 [all...] |
H A D | args.h | 192 Arg<string> input_shapes; member in struct:toco::ParsedModelFlags
|
/external/tensorflow/tensorflow/python/framework/ |
H A D | 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,
|
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
H A D | test_util.cc | 156 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 D | optional_tensor_test.cc | 36 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 D | lstm_test.cc | 38 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 D | unidirectional_sequence_lstm_test.cc | 40 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 D | 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)); 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 D | 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, 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 D | 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); 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 D | 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 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...] |