| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_concat_op.cc | 155 const TensorShapeList& input_shapes) { in Compute()
260 MklShapeList input_shapes(N); in Compute() local
490 const MklShapeList& input_shapes) { in CallEigenVersion()
532 const TensorShapeList& input_shapes) { in CallEigenVersion()
588 std::vector<MklDnnShape> input_shapes(N); in Compute() local
785 const TensorShapeList& input_shapes) { in CallEigenVersion()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 60 std::vector<PartialTensorShape>* input_shapes) { in ReadPartialShapesFromShapeMap()
80 const Node* n, const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceShapeOrShapeNOp()
121 const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceRankOp()
137 const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceSizeOp()
181 std::vector<PartialTensorShape> input_shapes; in MaybeReplaceShapeOp() local
|
| D | shape_refiner.cc | 215 std::vector<ShapeHandle> input_shapes(node->num_inputs()); in AddNode() local
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_show_multi.cc | 171 std::map<int, std::vector<int64>> input_shapes; in FormatInputShapes() local
|
| D | tfprof_node.h | 705 const std::map<int, std::vector<int64>> input_shapes() const { in input_shapes() function
706 std::map<int, std::vector<int64>> input_shapes; in input_shapes() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | cpp_shape_inference.cc | 75 std::vector<TensorShapeProto> input_shapes; in RunCppShapeInferenceImpl() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 54 const auto& input_shapes = input_->output_shapes(); in Dataset() local
|
| D | padded_batch_dataset_op.cc | 112 const auto& input_shapes = input_->output_shapes(); in Dataset() local
|
| D | tensor_queue_dataset_op.cc | 172 const auto& input_shapes = dataset()->input_->output_shapes(); in GetNextInternal() local
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | tfprof_output.proto | 70 map<int32, TensorShapeProto> input_shapes = 16; field
|
| D | tfprof_log.proto | 77 map<int32, Tuple> input_shapes = 16; field
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| D | model_cmdline_flags.cc | 296 std::vector<string> input_shapes = in ReadModelFlagsFromCommandLineFlags() local
|
| D | args.h | 192 Arg<string> input_shapes; member
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 34 const std::vector<TensorShapeProto>& input_shapes, in InferenceContext()
87 const std::vector<PartialTensorShape>& input_shapes, in InferenceContext()
138 const std::vector<ShapeHandle>& input_shapes, in InferenceContext()
1025 std::vector<string> input_shapes; in AttachContext() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | test_util.cc | 156 std::vector<std::vector<int>> input_shapes) { in BuildInterpreter()
|
| D | optional_tensor_test.cc | 36 const std::vector<std::vector<int>>& input_shapes) in LSTMOpModel()
|
| D | unidirectional_sequence_lstm_test.cc | 40 const std::vector<std::vector<int>>& input_shapes) in UnidirectionalLSTMOpModel()
|
| D | lstm_test.cc | 38 const std::vector<std::vector<int>>& input_shapes) in LSTMOpModel()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 431 std::vector<int>* input_mapping, std::vector<xla::Shape>* input_shapes, in BuildArguments()
|