| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph.py | 31 def function_def_to_graph(fdef, input_shapes=None): argument
92 def function_def_to_graph_def(fdef, input_shapes=None): argument
|
| D | cpp_shape_inference.cc | 75 std::vector<TensorShapeProto> input_shapes; in RunCppShapeInferenceImpl() local
|
| /external/tensorflow/tensorflow/lite/testing/model_coverage/ |
| D | model_coverage_lib.py | 256 input_shapes=None, argument
321 input_shapes=None, argument
349 input_shapes=None, argument
417 input_shapes=None, argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite.py | 477 input_shapes=None): argument
569 input_shapes=None, argument
607 input_shapes=None, argument
837 input_shapes=None): argument
848 input_shapes=None, argument
863 input_shapes=None, argument
|
| D | convert_saved_model.py | 222 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
|
| D | convert_saved_model_test.py | 163 input_shapes=None, argument
|
| D | convert.py | 237 input_shapes=None, argument
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ |
| D | utils.h | 31 std::vector<TensorShape> input_shapes; member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 62 std::vector<PartialTensorShape>* input_shapes) { in ReadPartialShapesFromShapeMap()
85 const Node* n, const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceShapeOrShapeNOp()
126 const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceRankOp()
142 const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceSizeOp()
186 std::vector<PartialTensorShape> input_shapes; in MaybeReplaceShapeOp() local
|
| /external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/ |
| D | trt_shfn.cc | 37 std::vector<tensorflow::TensorShape> input_shapes; in TRTEngineOpShapeInference() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_n_test.cc | 33 std::vector<std::vector<int>> input_shapes; in BaseAddNOpModel() local
|
| D | test_util.cc | 96 void SingleOpModel::BuildInterpreter(std::vector<std::vector<int>> input_shapes, in BuildInterpreter()
|
| D | lstm_test.cc | 40 const std::vector<std::vector<int>>& input_shapes, in LSTMOpModel()
275 const std::vector<std::vector<int>>& input_shapes, in HybridLSTMOpModel()
1637 const std::vector<std::vector<int>>& input_shapes, in LayerNormLSTMOpModel()
1652 const std::vector<std::vector<int>>& input_shapes, in HybridLayerNormLSTMOpModel()
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils_test.py | 34 input_shapes = [ variable
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization_test.cc | 450 std::vector<PartialTensorShape> input_shapes({{}}); in TEST() local
470 std::vector<PartialTensorShape> input_shapes({{-1, 2}}); in TEST() local
490 std::vector<PartialTensorShape> input_shapes({{1}}); in TEST() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 283 std::vector<TensorShape>* input_shapes, in AnalyzeInputs()
317 const std::vector<TensorShape>& input_shapes, in ConstructScopedAllocatorNode()
438 const std::vector<TensorShape>& input_shapes, in BuildSplitNode()
589 std::vector<TensorShape> input_shapes; in Rewrite() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_concat_op.cc | 63 const TensorShapeList& input_shapes) { in Compute()
497 memory::format FindMklCommonFormat(const MklDnnShapeList& input_shapes, in FindMklCommonFormat()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_show_multi.cc | 171 std::map<int, std::vector<int64>> input_shapes; in FormatInputShapes() 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/lite/experimental/microfrontend/ |
| D | audio_microfrontend_test.cc | 42 const std::vector<std::vector<int>>& input_shapes) in MicroFrontendOpModel()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 372 std::vector<TensorShape> input_shapes; in ComputeAsync() local
520 const std::vector<TensorShape>& input_shapes, OpKernelContext* ctx) { in GetEngine()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model_cmdline_flags.cc | 319 std::vector<string> input_shapes = in ReadModelFlagsFromCommandLineFlags() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 65 const auto& input_shapes = input_->output_shapes(); in Dataset() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.cc | 34 const std::vector<TensorShapeProto>& input_shapes, in InferenceContext()
89 const std::vector<PartialTensorShape>& input_shapes, in InferenceContext()
142 const std::vector<ShapeHandle>& input_shapes, in InferenceContext()
1152 std::vector<string> input_shapes; in AttachContext() local
|