| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph.py | 38 def function_def_to_graph(fdef, input_shapes=None): argument
130 def function_def_to_graph_def(fdef, input_shapes=None): argument
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_dnn_correctness_test.py | 63 input_shapes=None): argument
148 input_shapes=None): argument
188 input_shapes=None): argument
234 def __init__(self, initial_weights, input_shapes): argument
264 input_shapes=None): argument
|
| D | keras_embedding_model_correctness_test.py | 37 input_shapes=None): argument
85 input_shapes=None): argument
|
| D | keras_stateful_lstm_model_correctness_test.py | 54 input_shapes=None): argument
|
| D | keras_rnn_model_correctness_test.py | 47 input_shapes=None): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | einsum_op_test.py | 42 def _check(self, s, *input_shapes, **kwargs):
254 def check(equation, input_shapes, output_shape): argument
270 def check(equation, input_shapes, output_shape): argument
294 def _check_gradient(self, s, *input_shapes):
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_mlir_translate.cc | 48 const std::vector<llvm::Optional<std::vector<int>>>& input_shapes, in GraphdefToMlirImport()
106 const std::vector<llvm::Optional<std::vector<int>>>& input_shapes, in GraphdefToMlirTranslateFunction()
126 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToMlirTranslateFunction()
220 const std::vector<llvm::Optional<std::vector<int>>>& input_shapes, in GraphdefToSplattedMlirTranslateFunction()
270 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate_cl.cc | 38 opt<std::string> input_shapes( variable
|
| /external/tensorflow/tensorflow/lite/testing/model_coverage/ |
| D | model_coverage_lib.py | 590 input_shapes=None, argument
672 input_shapes=None, argument
715 input_shapes=None, argument
882 input_shapes=None, argument
912 input_shapes=None, argument
949 input_shapes=None, argument
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 114 std::vector<shape_inference::ShapeHandle> input_shapes; in TEST() local
133 std::vector<shape_inference::ShapeHandle> input_shapes; in TEST() local
151 std::vector<shape_inference::ShapeHandle> input_shapes; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_instance.proto | 10 repeated TensorShapeProto input_shapes = 1; field
|
| D | trt_lru_cache.cc | 100 const std::vector<TensorShape>& input_shapes) { in GetEngineContext()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 64 std::vector<PartialTensorShape>* input_shapes) { in ReadPartialShapesFromShapeMap()
87 const Node* n, const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceShapeOrShapeNOp()
128 const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceRankOp()
144 const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceSizeOp()
188 std::vector<PartialTensorShape> input_shapes; in MaybeReplaceShapeOp() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | concatenation.h | 29 const RuntimeShape* const* input_shapes, in Concatenation()
75 const RuntimeShape* const* input_shapes, in ConcatenationWithScaling()
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | layer_correctness_test.py | 66 def _create_model_from_layer(self, layer, input_shapes): argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite.py | 1504 input_shapes=None, argument
1846 input_shapes=None): argument
1938 input_shapes=None, argument
1986 input_shapes=None, argument
2047 input_shapes=None): argument
2058 input_shapes=None, argument
2073 input_shapes=None, argument
|
| D | convert_saved_model.py | 155 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
|
| D | lite_v2_test_util.py | 37 def _evaluateTFLiteModel(self, tflite_model, input_data, input_shapes=None): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | einsum_op.cc | 67 std::vector<TensorShape> input_shapes; in Compile() local
|
| D | sort_ops.cc | 69 std::vector<TensorShape> input_shapes; in Compile() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | add_n_test.cc | 33 std::vector<std::vector<int>> input_shapes; in BaseAddNOpModel() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_n_test.cc | 34 std::vector<std::vector<int>> input_shapes; in BaseAddNOpModel() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_executable_info.proto | 30 repeated xla.ShapeProto input_shapes = 2; field
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | node_shader.h | 107 std::vector<std::array<int64_t, 4>> input_shapes; member
|
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper_pybind11.cc | 37 [](CalibrationWrapper& self, py::handle& input_shapes) { in PYBIND11_MODULE()
|