| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph.py | 33 def function_def_to_graph(fdef, input_shapes=None): argument
122 def function_def_to_graph_def(fdef, input_shapes=None): argument
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_dnn_correctness_test.py | 59 input_shapes=None): argument
145 input_shapes=None): argument
188 input_shapes=None): argument
237 def __init__(self, initial_weights, input_shapes): argument
266 input_shapes=None): argument
|
| D | keras_embedding_model_correctness_test.py | 37 input_shapes=None): argument
89 input_shapes=None): argument
|
| D | keras_stateful_lstm_model_correctness_test.py | 56 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 | 40 def _check(self, s, *input_shapes, **kwargs):
248 def check(equation, input_shapes, output_shape): argument
265 def check(equation, input_shapes, output_shape): argument
287 def _check_gradient(self, s, *input_shapes):
|
| /external/tensorflow/tensorflow/lite/testing/model_coverage/ |
| D | model_coverage_lib.py | 391 input_shapes=None, argument
459 input_shapes=None, argument
502 input_shapes=None, argument
622 input_shapes=None, argument
659 input_shapes=None, argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_instance.proto | 10 repeated TensorShapeProto input_shapes = 1; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | optimize_input_output_buffer_alias.cc | 41 absl::Span<const Shape* const> input_shapes, const Shape& output_shape, in Build()
100 std::vector<const Shape*> input_shapes; in Run() local
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 118 std::vector<shape_inference::ShapeHandle> input_shapes; in TEST() local
137 std::vector<shape_inference::ShapeHandle> input_shapes; in TEST() local
155 std::vector<shape_inference::ShapeHandle> input_shapes; in TEST() local
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite.py | 713 input_shapes=None): argument
805 input_shapes=None, argument
846 input_shapes=None, argument
1166 input_shapes=None): argument
1177 input_shapes=None, argument
1192 input_shapes=None, argument
|
| D | convert_saved_model.py | 155 def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | concat.cc | 114 const ConcatAttributes& attr, const std::vector<BHWC>& input_shapes) { in ConcatZ()
171 const ConcatAttributes& attr, const std::vector<BHWC>& input_shapes) { in ConcatX()
261 const ConcatAttributes& attr, const std::vector<BHWC>& input_shapes) { in ConcatY()
352 const ConcatAttributes& attr, const std::vector<BHWC>& input_shapes) { in Concat()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_mlir_translate.cc | 49 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToMlirImport()
103 absl::string_view input_shapes, absl::string_view output_arrays, in GraphdefToMlirTranslateFunction()
164 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/kernels/ |
| D | test_util.cc | 146 void SingleOpModel::BuildInterpreter(std::vector<std::vector<int>> input_shapes, in BuildInterpreter()
217 std::vector<std::vector<int>> input_shapes) { in BuildInterpreter()
223 void SingleOpModel::BuildInterpreter(std::vector<std::vector<int>> input_shapes, in BuildInterpreter()
230 void SingleOpModel::BuildInterpreter(std::vector<std::vector<int>> input_shapes, in BuildInterpreter()
|
| D | add_n_test.cc | 33 std::vector<std::vector<int>> input_shapes; in BaseAddNOpModel() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | concatenation.h | 28 const RuntimeShape* const* input_shapes, in Concatenation()
75 const RuntimeShape* const* input_shapes, in ConcatenationWithScaling()
|
| /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/python/keras/mixed_precision/experimental/ |
| D | layer_correctness_test.py | 66 def _create_model_from_layer(self, layer, input_shapes): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | einsum_op.cc | 67 std::vector<TensorShape> input_shapes; in Compile() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 465 def _einsum_v1_parse_and_resolve_equation(equation, input_shapes): argument
873 def _einsum_v2_parse_and_resolve_equation(equation, input_shapes): argument
|
| D | control_flow_util_v2.py | 264 def get_func_graph(op, input_shapes, func_name): argument
|
| D | special_math_ops_test.py | 438 def _check(self, s, *input_shapes, **kwargs):
738 def check(equation, input_shapes, output_shape): argument
761 def _check_gradient(self, s, *input_shapes):
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | tf_to_tfl_flatbuffer.cc | 54 absl::string_view input_dtypes, absl::string_view input_shapes, in LoadFromGraphdefOrMlirSource()
|