Searched refs:output_dim0 (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseSegmentMeanGrad.pbtxt | 22 name: "output_dim0"
30 value is output_dim0.
|
| D | api_def_SparseSegmentSqrtNGrad.pbtxt | 22 name: "output_dim0"
30 value is output_dim0.
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_test.cc | 129 Tensor output_dim0(DT_INT32, TensorShape({})); in SparseSegmentMeanGradHelper() local
130 output_dim0.scalar<int32>()() = kUniqueIndices; in SparseSegmentMeanGradHelper()
147 .Input(test::graph::Constant(g, output_dim0)) in SparseSegmentMeanGradHelper()
|
| D | segment_reduction_ops.cc | 1015 const Tensor& output_dim0 = context->input(3); in Compute() local
1021 OP_REQUIRES(context, IsLegacyScalar(output_dim0.shape()), in Compute()
1030 internal::SubtleMustCopy(output_dim0.scalar<SegmentId>()()); in Compute()
|
| D | batch_kernels.cc | 55 int64 output_dim0 = 0; in Concat() local
76 output_dim0 += input.dim_size(0); in Concat()
80 output_shape.set_dim(0, output_dim0); in Concat()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 3529 …argspec: "args=[\'grad\', \'indices\', \'segment_ids\', \'output_dim0\', \'name\'], varargs=None, …
3541 …argspec: "args=[\'grad\', \'indices\', \'segment_ids\', \'output_dim0\', \'name\'], varargs=None, …
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 3529 …argspec: "args=[\'grad\', \'indices\', \'segment_ids\', \'output_dim0\', \'name\'], varargs=None, …
3541 …argspec: "args=[\'grad\', \'indices\', \'segment_ids\', \'output_dim0\', \'name\'], varargs=None, …
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 32831 …cope *Scope, grad tf.Output, indices tf.Output, segment_ids tf.Output, output_dim0 tf.Output) (out…
32838 grad, indices, segment_ids, output_dim0,
32944 …cope *Scope, grad tf.Output, indices tf.Output, segment_ids tf.Output, output_dim0 tf.Output) (out…
32951 grad, indices, segment_ids, output_dim0,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 34160 name: "output_dim0"
34193 name: "output_dim0"
34310 name: "output_dim0"
34343 name: "output_dim0"
|
| D | ops_history.v2.pbtxt | 77232 name: "output_dim0"
77379 name: "output_dim0"
|
| D | ops_history.v1.pbtxt | 77859 name: "output_dim0"
78006 name: "output_dim0"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 36974 name: "output_dim0"
37121 name: "output_dim0"
|