Searched refs:GetOrCreateAdd (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | lrn_ops.cc | 56 *ctx->GetOrCreateAdd(accumulation_type), in Compile()
143 *ctx->GetOrCreateAdd(accumulation_type), in Compile()
160 *ctx->GetOrCreateAdd(accumulation_type), in Compile()
|
| D | fake_quantize_ops.cc | 220 *ctx->GetOrCreateAdd(accumulation_type)); in Compile()
229 *ctx->GetOrCreateAdd(accumulation_type)); in Compile()
350 *ctx->GetOrCreateAdd(accumulation_type), reduce_axes); in Compile()
359 *ctx->GetOrCreateAdd(accumulation_type), reduce_axes); in Compile()
|
| D | softmax_op.cc | 71 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in CrossEntropyWithLogits()
89 *ctx->GetOrCreateAdd(accumulation_type), {kClassDim}); in CrossEntropyWithLogits()
|
| D | l2loss_op.cc | 42 *ctx->GetOrCreateAdd(accumulation_type), dims); in Compile()
|
| D | lower_upper_bound_ops.cc | 78 *ctx->GetOrCreateAdd(accumulation_type), {2}); in BuildLowerUpperBoundOp()
|
| D | all_reduce_op.cc | 62 return ctx->GetOrCreateAdd(dtype); in Compile()
|
| D | batch_norm_op.cc | 319 *ctx->GetOrCreateAdd(accumulation_type), reduction_dims); in Compile()
332 *ctx->GetOrCreateAdd(accumulation_type), reduction_dims); in Compile()
|
| D | scan_ops.cc | 98 reducer = ctx->GetOrCreateAdd(dtype); in Compile()
|
| D | image_resize_ops.cc | 630 w_weight, scalar_zero_op, *ctx->GetOrCreateAdd(output_dtype), {1}); in GeneralCompile()
649 h_weight, scalar_zero_op, *ctx->GetOrCreateAdd(output_dtype), {1}); in GeneralCompile()
|
| D | image_ops.cc | 207 *context->GetOrCreateAdd(accumulation_type), in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.h | 98 const xla::XlaComputation* GetOrCreateAdd(const DataType type);
|
| D | xla_context.cc | 117 const xla::XlaComputation* XlaContext::GetOrCreateAdd(const DataType type) { in GetOrCreateAdd() function in tensorflow::XlaContext
|
| D | xla_op_kernel.h | 336 const xla::XlaComputation* GetOrCreateAdd(const DataType type);
|
| D | xla_op_kernel.cc | 785 const xla::XlaComputation* XlaOpKernelContext::GetOrCreateAdd( in GetOrCreateAdd() function in tensorflow::XlaOpKernelContext
787 return xla_context()->GetOrCreateAdd(type); in GetOrCreateAdd()
|