| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A 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()
|
| H A D | tf_mlir_translate_cl.cc | 39 opt<std::string> input_shapes( variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| H A 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):
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/test_op/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/kernels/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | function_def_to_graph.py | 37 input_shapes=None): argument
149 def function_def_to_graph_def(fdef, input_shapes=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/optimize/ |
| H A 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()
|
| H A D | calibration_wrapper_pybind11.cc | 58 [](CalibrationWrapper& self, py::handle& input_shapes) { in PYBIND11_MODULE()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops_test.cc | 790 Status CreateConcatTensorGraph(absl::Span<const TensorShape> input_shapes, in CreateConcatTensorGraph()
821 absl::Span<const TensorShape> input_shapes, in CreateConcatResourceGraph()
1017 const std::vector<TensorShape> input_shapes{{1}, {2}}; in TEST_P() local
1086 const std::vector<TensorShape> input_shapes{{2, 2}, {2, 2}, {2, 2}, {2, 2}}; in TEST_P() local
1105 const std::vector<TensorShape> input_shapes{{2, 2}, {2, 2}, {2, 2}, {2, 2}}; in TEST_P() local
1123 const std::vector<TensorShape> input_shapes{{2, 2}, {2, 2}, {2, 2}, {2, 2}}; in TEST_P() local
1168 const std::vector<TensorShape> input_shapes(num_inputs, base_input_shape); in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | concatenation.h | 31 const RuntimeShape* const* input_shapes, in Concatenation()
77 const RuntimeShape* const* input_shapes, in ConcatenationWithScaling()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | special_math_ops_test.py | 637 def _check(self, s, *input_shapes, **kwargs):
930 def check(equation, input_shapes, output_shape): argument
953 def _check_gradient(self, s, *input_shapes):
|
| H A 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
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/UNIT/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | scoped_allocator_optimizer.cc | 451 std::vector<TensorShape>* input_shapes, in AnalyzeInputs()
514 const std::vector<TensorShape>& input_shapes, in ConstructScopedAllocatorNode()
689 const std::vector<TensorShape>& input_shapes, in BuildSplitNode()
842 std::vector<TensorShape> input_shapes; in Rewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | test_util.cc | 177 void SingleOpModel::BuildInterpreter(std::vector<std::vector<int>> input_shapes, in BuildInterpreter()
292 std::vector<std::vector<int>> input_shapes) { in BuildInterpreter()
|
| H A D | add_n_test_common.h | 29 std::vector<std::vector<int>> input_shapes; in BaseAddNOpModel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | sharding_util_ops.cc | 317 std::vector<TensorShape> input_shapes; in CompileInternal() local
399 std::vector<TensorShape>& input_shapes, in GetInputsAndOutputShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| H A D | xla_ops.cc | 712 std::vector<shape_inference::ShapeHandle> input_shapes; in __anon6b9e6b800902() local
938 std::vector<shape_inference::ShapeHandle> input_shapes; in __anon6b9e6b800d02() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| H A D | trt_engine_instance.proto | 10 repeated TensorShapeProto input_shapes = 1; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_ops.cc | 90 std::vector<PartialTensorShape> input_shapes; in __anon3c2861b20202() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | shape_inference.cc | 36 const std::vector<PartialTensorShape>& input_shapes, in InferenceContext()
88 const std::vector<ShapeHandle>& input_shapes, in InferenceContext()
1158 std::vector<string> input_shapes; in AttachContext() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_graph.cc | 356 std::vector<PartialTensorShape> input_shapes; in CreateTRTNode() local
694 const std::vector<PartialTensorShape>& input_shapes, in CreateStaticEngine()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/consolidate_attrs/ |
| H A D | pass.cc | 163 ArrayAttr input_shapes = in reifyAndDropFunctionArgumentAttributes() local
404 SmallVector<Attribute> input_shapes, arg_attrs, res_attrs; in prepareFunctionAttributes() local
|