| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | split_op.cc | 46 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(0, &split_dim_orig)); in Compile()
109 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(2, &split_dim_orig)); in Compile()
|
| D | concat_op.cc | 54 ctx->ConstantInputAsIntScalar(axis_index_, &concat_dim)); in Compile()
162 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(0, &cdim)); in Compile()
|
| D | topk_op.cc | 35 OP_REQUIRES_OK(context, context->ConstantInputAsIntScalar(1, &k)); in Compile()
|
| D | one_hot_op.cc | 60 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &depth)); in Compile()
|
| D | index_ops.cc | 46 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &dim)); in Compile()
|
| D | xla_conv_op.cc | 57 OP_REQUIRES_OK(context, context->ConstantInputAsIntScalar( in Compile()
|
| D | scan_ops.cc | 60 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &axis)); in Compile()
|
| D | index_ops_cpu.cc | 54 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &dim)); in Compile()
|
| D | segment_reduction_ops.cc | 60 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(2, &num_segments)); in Compile()
|
| D | tensor_list_ops.cc | 106 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &num_elements)); in Compile()
138 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &max_num_elements)); in Compile()
|
| D | categorical_op.cc | 45 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(1, &num_samples)); in Compile()
|
| D | sequence_ops.cc | 141 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar("num", &num)); in Compile()
|
| D | stack_ops.cc | 100 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(0, &max_size)); in Compile()
|
| D | gather_op.cc | 168 OP_REQUIRES_OK(context, context->ConstantInputAsIntScalar(2, &axis)); in Compile()
|
| D | tensor_array_ops.cc | 150 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntScalar(0, &size)); in Compile()
|
| D | image_ops.cc | 434 OP_REQUIRES_OK(context, context->ConstantInputAsIntScalar(2, &output_size)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 117 Status ConstantInputAsIntScalar(int index, int64* out);
118 Status ConstantInputAsIntScalar(absl::string_view name, int64* out);
|
| D | xla_op_kernel.cc | 210 Status XlaOpKernelContext::ConstantInputAsIntScalar(int index, int64* out) { in ConstantInputAsIntScalar() function in tensorflow::XlaOpKernelContext
216 Status XlaOpKernelContext::ConstantInputAsIntScalar(absl::string_view name, in ConstantInputAsIntScalar() function in tensorflow::XlaOpKernelContext
219 return ConstantInputAsIntScalar(index, out); in ConstantInputAsIntScalar()
|