Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:ConstantInputAsIntVector (Results 1 - 12 of 12) sorted by null

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
slice_op.cc	58 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(2, &size)); 59 if (ctx->ConstantInputAsIntVector(1, &begin).ok()) {
fft_ops.cc	53 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &fft_length));
reverse_op.cc	95 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &axes));
transpose_op.cc	115 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(0, &perm));
bcast_ops.cc	48 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(i, &shape));
pooling_ops.cc	95 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &ksize)); 107 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(2, &stride)); 298 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_)); 305 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(4, &stride_));
tensor_array_ops.cc	317 Status status = ctx->ConstantInputAsIntVector(1, &const_indices); 388 Status status = ctx->ConstantInputAsIntVector(1, &const_indices); 487 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(2, &lengths));
batchtospace_op.cc	153 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &block_shape));
spacetobatch_op.cc	156 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &block_shape));
image_resize_ops.cc	276 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &out_size));
/external/tensorflow/tensorflow/compiler/tf2xla/
xla_op_kernel.h	114 Status ConstantInputAsIntVector(int index, std::vector<int64>* out);
xla_op_kernel.cc	220 Status XlaOpKernelContext::ConstantInputAsIntVector(int index,

Completed in 106 milliseconds