| /external/tflite-support/tensorflow_lite_support/custom_ops/python/ |
| D | tflite_text_api.py | 57 reduction_type=None, argument
87 if reduction_type is not tf_text.Reduction.STRING_JOIN:
92 if reduction_type is tf_text.Reduction.STRING_JOIN and axis != -1:
116 return tf_text.ngrams(ragged_tensor, width, axis, reduction_type,
123 return tf_text.ngrams(data, width, axis, reduction_type, string_separator,
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | ngrams_test.py | 78 input_tensor, width, reduction_type=tf_text.Reduction.STRING_JOIN)
94 input_tensor, width, reduction_type=tf_text.Reduction.STRING_JOIN)
114 input_tensor, 2, reduction_type=tf_text.Reduction.STRING_JOIN)
134 input_tensor, 3, reduction_type=tf_text.Reduction.STRING_JOIN)
153 input_tensor, 2, reduction_type=tf_text.Reduction.STRING_JOIN)
184 input_tensor, 3, reduction_type=tf_text.Reduction.STRING_JOIN)
|
| D | ngrams.cc | 59 std::string reduction_type; member
65 reduction_type(m["reduction_type"].ToString()), in NgramsAttributes()
87 TF_LITE_ENSURE(context, attributes.reduction_type == kStringJoin); in Prepare()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | typed_conditional_accumulator_base.h | 39 const string& reduction_type) in TypedConditionalAccumulatorBase() argument
40 : ConditionalAccumulatorBase(dtype, shape, name, reduction_type) {} in TypedConditionalAccumulatorBase()
|
| D | conditional_accumulator.h | 56 const string& name, const string& reduction_type) in ConditionalAccumulator() argument
58 reduction_type) {} in ConditionalAccumulator()
|
| D | conditional_accumulator_base.cc | 23 const string& reduction_type) in ConditionalAccumulatorBase() argument
27 reduction_type_(reduction_type) { in ConditionalAccumulatorBase()
|
| D | conditional_accumulator_base.h | 55 const string& name, const string& reduction_type);
|
| D | sparse_conditional_accumulator.h | 53 const string& name, const string& reduction_type) in SparseConditionalAccumulator() argument
56 dtype, shape, name, reduction_type) { in SparseConditionalAccumulator()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | tftext_utils.cc | 238 auto reduction_type = in CreateNgramsCustomOption() local
240 if (!reduction_type) { in CreateNgramsCustomOption()
246 std::string reduction_type_str(reduction_type.getValue().data(), in CreateNgramsCustomOption()
247 reduction_type.getValue().size()); in CreateNgramsCustomOption()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduction_ops_common.cc | 31 DataType reduction_type) in XlaReductionOp() argument
32 : XlaOpKernel(ctx), reduction_type_(reduction_type) { in XlaReductionOp()
|
| D | reduction_ops.h | 36 XlaReductionOp(OpKernelConstruction* ctx, DataType reduction_type);
|
| D | pooling_ops.cc | 44 const DataType reduction_type) in PoolingOp() argument
47 reduction_type_(reduction_type) { in PoolingOp()
445 auto reduction_type = XlaHelpers::SumAccumulationType(ctx->input_type(1)); in Compile() local
447 ctx, DataTypeToPrimitiveType(reduction_type, &xla_reduction_type)); in Compile()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ResourceConditionalAccumulator.pbtxt | 51 name: "reduction_type"
|
| D | ConditionalAccumulator.pbtxt | 256 name: "reduction_type"
|
| D | SparseConditionalAccumulator.pbtxt | 256 name: "reduction_type"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ResourceConditionalAccumulator.pbtxt | 51 name: "reduction_type"
|
| D | ConditionalAccumulator.pbtxt | 256 name: "reduction_type"
|
| D | SparseConditionalAccumulator.pbtxt | 256 name: "reduction_type"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-conditional-accumulator.pbtxt | 20 …argspec: "args=[\'self\', \'dtype\', \'shape\', \'shared_name\', \'name\', \'reduction_type\'], va…
|
| D | tensorflow.-sparse-conditional-accumulator.pbtxt | 20 …argspec: "args=[\'self\', \'dtype\', \'shape\', \'shared_name\', \'name\', \'reduction_type\'], va…
|
| D | tensorflow.sparse.-sparse-conditional-accumulator.pbtxt | 20 …argspec: "args=[\'self\', \'dtype\', \'shape\', \'shared_name\', \'name\', \'reduction_type\'], va…
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conditional_accumulator_test.py | 46 dtypes_lib.float32, name="Q", reduction_type="Invalid")
259 reduction_type="SUM")
287 reduction_type="Invalid")
340 reduction_type="SUM")
|
| D | sparse_conditional_accumulator_test.py | 79 dtypes_lib.float32, name="Q", reduction_type="Invalid")
208 dtypes_lib.float32, name="Q", shape=(), reduction_type="SUM")
229 dtypes_lib.float32, name="Q", shape=(), reduction_type="Invalid")
315 reduction_type="SUM")
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 1335 reduction_type="MEAN"): argument
1351 reduction_type=reduction_type)
1437 reduction_type="MEAN"): argument
1443 reduction_type=reduction_type)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 833 …argspec: "args=[\'dtype\', \'shape\', \'container\', \'shared_name\', \'reduction_type\', \'name\'…
3613 …argspec: "args=[\'dtype\', \'shape\', \'container\', \'shared_name\', \'reduction_type\', \'name\'…
4269 …argspec: "args=[\'dtype\', \'shape\', \'container\', \'shared_name\', \'reduction_type\', \'name\'…
|