HomeSort by relevance Sort by last modified time
    Searched refs:GetOrCreateAdd (Results 1 - 14 of 14) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
l2loss_op.cc 39 const xla::Computation& add = *ctx->GetOrCreateAdd(dtype);
lrn_ops.cc 53 *ctx->GetOrCreateAdd(input_type(0)),
136 *ctx->GetOrCreateAdd(input_type(0)),
151 *ctx->GetOrCreateAdd(input_type(0)),
bias_ops.cc 108 *ctx->GetOrCreateAdd(input_type(0)), reduce_dims);
scan_ops.cc 99 reducer = ctx->GetOrCreateAdd(dtype);
batch_norm_op.cc 170 *ctx->GetOrCreateAdd(scale_dtype), reduction_dims);
180 XlaHelpers::Zero(b, scale_dtype), *ctx->GetOrCreateAdd(scale_dtype),
softmax_op.cc 47 const xla::Computation& add_func = *ctx->GetOrCreateAdd(type);
85 const xla::Computation& add_func = *ctx->GetOrCreateAdd(type);
pooling_ops.cc 219 *ctx->GetOrCreateAdd(dtype), window_ksize, window_stride,
238 return ctx->GetOrCreateAdd(dtype);
493 padded_gradients, zero, *ctx->GetOrCreateAdd(dtype), ksize_,
fake_quantize_ops.cc 270 *ctx->GetOrCreateAdd(data_type));
276 *ctx->GetOrCreateAdd(data_type));
conv_ops.cc 170 *ctx->GetOrCreateAdd(dtype),
image_ops.cc 184 /*computation=*/*context->GetOrCreateAdd(type),
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_context.h 109 const xla::Computation* GetOrCreateAdd(const DataType type);
xla_op_kernel.h 218 const xla::Computation* GetOrCreateAdd(const DataType type);
xla_context.cc 154 const xla::Computation* XlaContext::GetOrCreateAdd(const DataType type) {
xla_op_kernel.cc 470 const xla::Computation* XlaOpKernelContext::GetOrCreateAdd(
472 return XlaContext::Get(context_).GetOrCreateAdd(type);

Completed in 413 milliseconds