Home
last modified time | relevance | path

Searched refs:reduction_type (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtyped_conditional_accumulator_base.h39 const string& reduction_type) in TypedConditionalAccumulatorBase() argument
40 : ConditionalAccumulatorBase(dtype, shape, name, reduction_type) {} in TypedConditionalAccumulatorBase()
Dconditional_accumulator.h56 const string& name, const string& reduction_type) in ConditionalAccumulator() argument
58 reduction_type) {} in ConditionalAccumulator()
Dconditional_accumulator_base.cc23 const string& reduction_type) in ConditionalAccumulatorBase() argument
27 reduction_type_(reduction_type) { in ConditionalAccumulatorBase()
Dconditional_accumulator_base.h55 const string& name, const string& reduction_type);
Dsparse_conditional_accumulator.h53 const string& name, const string& reduction_type) in SparseConditionalAccumulator() argument
56 dtype, shape, name, reduction_type) { in SparseConditionalAccumulator()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduction_ops_common.cc31 DataType reduction_type) in XlaReductionOp() argument
32 : XlaOpKernel(ctx), reduction_type_(reduction_type) { in XlaReductionOp()
Dreduction_ops.h36 XlaReductionOp(OpKernelConstruction* ctx, DataType reduction_type);
Dpooling_ops.cc42 const DataType reduction_type) in PoolingOp() argument
45 reduction_type_(reduction_type) { in PoolingOp()
424 auto reduction_type = XlaHelpers::SumAccumulationType(ctx->input_type(1)); in Compile() local
426 ctx, DataTypeToPrimitiveType(reduction_type, &xla_reduction_type)); in Compile()
/external/tensorflow/tensorflow/python/kernel_tests/
Dconditional_accumulator_test.py60 dtypes_lib.float32, name="Q", reduction_type="Invalid")
294 reduction_type="SUM")
322 reduction_type="Invalid")
375 reduction_type="SUM")
Dsparse_conditional_accumulator_test.py79 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")
309 reduction_type="SUM")
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-conditional-accumulator.pbtxt20 …argspec: "args=[\'self\', \'dtype\', \'shape\', \'shared_name\', \'name\', \'reduction_type\'], va…
Dtensorflow.sparse.-sparse-conditional-accumulator.pbtxt20 …argspec: "args=[\'self\', \'dtype\', \'shape\', \'shared_name\', \'name\', \'reduction_type\'], va…
Dtensorflow.-sparse-conditional-accumulator.pbtxt20 …argspec: "args=[\'self\', \'dtype\', \'shape\', \'shared_name\', \'name\', \'reduction_type\'], va…
Dtensorflow.raw_ops.pbtxt633 …argspec: "args=[\'dtype\', \'shape\', \'container\', \'shared_name\', \'reduction_type\', \'name\'…
3469 …argspec: "args=[\'dtype\', \'shape\', \'container\', \'shared_name\', \'reduction_type\', \'name\'…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.-sparse-conditional-accumulator.pbtxt20 …argspec: "args=[\'self\', \'dtype\', \'shape\', \'shared_name\', \'name\', \'reduction_type\'], va…
Dtensorflow.raw_ops.pbtxt633 …argspec: "args=[\'dtype\', \'shape\', \'container\', \'shared_name\', \'reduction_type\', \'name\'…
3469 …argspec: "args=[\'dtype\', \'shape\', \'container\', \'shared_name\', \'reduction_type\', \'name\'…
/external/tensorflow/tensorflow/python/ops/
Ddata_flow_ops.py1256 reduction_type="MEAN"): argument
1272 reduction_type=reduction_type)
1355 reduction_type="MEAN"): argument
1361 reduction_type=reduction_type)
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt6069 name: "reduction_type"
36259 name: "reduction_type"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt13775 name: "reduction_type"
75435 name: "reduction_type"
Dops_history.v1.pbtxt13833 name: "reduction_type"
76062 name: "reduction_type"