| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_dnn_correctness_test.py | 61 input_shapes=None): argument
146 input_shapes=None): argument
186 input_shapes=None): argument
232 def __init__(self, initial_weights, input_shapes): argument
262 input_shapes=None): argument
|
| D | keras_embedding_model_correctness_test.py | 34 input_shapes=None): argument
82 input_shapes=None): argument
|
| D | keras_stateful_lstm_model_correctness_test.py | 52 input_shapes=None): argument
|
| D | keras_rnn_model_correctness_test.py | 45 input_shapes=None): argument
|
| /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/lite/kernels/shim/test_op/ |
| D | simple_tflite_op_test.cc | 35 const std::vector<std::vector<int>>& input_shapes, in SimpleOpModel()
80 std::vector<std::vector<int>> input_shapes = {{}, {}, {2}}; in TEST() local
113 std::vector<std::vector<int>> input_shapes = {{}}; in TEST() local
|
| /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()
236 const std::vector<llvm::Optional<std::vector<int>>>& input_shapes, in GraphdefToSplattedMlirTranslateFunction()
286 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/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 | 78 def _create_model_from_layer(self, layer, input_shapes): argument
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite.py | 1891 input_shapes=None, argument
2263 input_shapes=None): argument
2355 input_shapes=None, argument
2403 input_shapes=None, argument
2464 input_shapes=None): argument
2475 input_shapes=None, argument
2490 input_shapes=None, argument
|
| D | convert_saved_model.py | 159 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_n_test.cc | 34 std::vector<std::vector<int>> input_shapes; in BaseAddNOpModel() local
|
| /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 | 822 def _einsum_v1_parse_and_resolve_equation(equation, input_shapes): argument
1234 def _einsum_v2_parse_and_resolve_equation(equation, input_shapes): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | sort_ops.cc | 69 std::vector<TensorShape> input_shapes; in Compile() local
|
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper_pybind11.cc | 49 [](CalibrationWrapper& self, py::handle& input_shapes) { in PYBIND11_MODULE()
|
| /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/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 709 std::vector<shape_inference::ShapeHandle> input_shapes; in __anon592c2f360902() local
935 std::vector<shape_inference::ShapeHandle> input_shapes; in __anon592c2f360d02() local
|