| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph.py | 37 input_shapes=None): argument
149 def function_def_to_graph_def(fdef, input_shapes=None): argument
|
| /external/tensorflow/tensorflow/lite/kernels/shim/test_op/ |
| D | simple_tflite_op_test.cc | 35 const std::vector<std::vector<int>>& input_shapes, in SimpleOpModel()
81 std::vector<std::vector<int>> input_shapes = {{}, {}, {2}}; in TEST() local
116 std::vector<std::vector<int>> input_shapes = {{}}; in TEST() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | einsum_op_test.py | 38 def _check(self, s, *input_shapes, **kwargs):
250 def check(equation, input_shapes, output_shape): argument
266 def check(equation, input_shapes, output_shape): argument
290 def _check_gradient(self, s, *input_shapes):
|
| /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/mlir/tensorflow/translate/ |
| D | tf_mlir_translate.cc | 48 const std::vector<llvm::Optional<std::vector<int>>>& input_shapes, in GraphdefToMlirImport()
108 const std::vector<llvm::Optional<std::vector<int>>>& input_shapes, in GraphdefToMlirTranslateFunction()
128 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToMlirTranslateFunction()
246 const std::vector<llvm::Optional<std::vector<int>>>& input_shapes, in GraphdefToSplattedMlirTranslateFunction()
297 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToSplattedMlirTranslateFunction()
|
| D | tf_mlir_translate_cl.cc | 39 opt<std::string> input_shapes( variable
|
| /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/lite/python/optimize/ |
| D | calibration_wrapper_pybind11.cc | 58 [](CalibrationWrapper& self, py::handle& input_shapes) { in PYBIND11_MODULE()
|
| D | calibration_wrapper.cc | 141 PyObject* input_shapes, size_t index) { in ConvertInputShapeToVector()
233 PyObject* CalibrationWrapper::Prepare(PyObject* input_shapes, in Prepare()
274 PyObject* CalibrationWrapper::Prepare(PyObject* input_shapes) { in Prepare()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 65 std::vector<PartialTensorShape>* input_shapes) { in ReadPartialShapesFromShapeMap()
88 const Node* n, const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceShapeOrShapeNOp()
129 const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceRankOp()
145 const std::vector<PartialTensorShape>& input_shapes, in MaybeReplaceSizeOp()
189 std::vector<PartialTensorShape> input_shapes; in MaybeReplaceShapeOp() local
|
| D | node_file_writer.cc | 137 AttrValue& input_shapes = (*def.mutable_attr())["_input_shapes"]; in RecordNodeExecution() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | concatenation.h | 31 const RuntimeShape* const* input_shapes, in Concatenation()
77 const RuntimeShape* const* input_shapes, in ConcatenationWithScaling()
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_ops.cc | 90 std::vector<PartialTensorShape> input_shapes; in __anonac88f2010202() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_n_test_common.h | 29 std::vector<std::vector<int>> input_shapes; in BaseAddNOpModel() local
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model.py | 155 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
|
| D | lite.py | 2242 input_shapes=None, argument
2620 input_shapes=None): argument
2713 input_shapes=None, argument
2765 input_shapes=None, argument
2830 input_shapes=None): argument
2841 input_shapes=None, argument
2856 input_shapes=None, argument
|
| D | convert_saved_model_test.py | 81 input_shapes=None, argument
|
| /external/ComputeLibrary/tests/validation/fixtures/UNIT/ |
| D | DynamicTensorFixture.h | 255 …void setup(std::vector<TensorShape> input_shapes, TensorShape weights_shape, TensorShape bias_shap… in setup()
394 void setup(std::vector<TensorShape> input_shapes) in setup()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_executable_info.proto | 30 repeated xla.ShapeProto input_shapes = 2; field
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 833 def _einsum_v1_parse_and_resolve_equation(equation, input_shapes): argument
1244 def _einsum_v2_parse_and_resolve_equation(equation, input_shapes): argument
|
| D | control_flow_util_v2.py | 296 def get_func_graph(op, input_shapes, func_name): argument
|
| /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/python/compiler/tensorrt/test/ |
| D | batch_matmul_test.py | 34 def BuildParams(self, graph_fn, dtype, input_shapes, output_shapes): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/ |
| D | tf_to_tfjs_json.cc | 77 absl::string_view input_dtypes, absl::string_view input_shapes, in LoadFromGraphdefOrMlirSource()
|