| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduce_window_op.cc | 45 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector( in Compile()
47 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("window_strides", in Compile()
49 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("base_dilations", in Compile()
51 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector( in Compile()
|
| D | xla_pad_op.cc | 40 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("padding_low", in Compile()
42 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("padding_high", in Compile()
44 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector( in Compile()
|
| D | xla_conv_op.cc | 53 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("window_strides", in Compile()
55 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("lhs_dilation", in Compile()
57 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("rhs_dilation", in Compile()
|
| D | listdiff_op.cc | 74 TF_RETURN_IF_ERROR(context->ConstantInputAsIntVector(0, &x_input)); in ListDiff()
75 TF_RETURN_IF_ERROR(context->ConstantInputAsIntVector(1, &y_input)); in ListDiff()
|
| D | bcast_ops.cc | 50 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(i, &shape)); in Compile()
102 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector( in Compile()
|
| D | xla_select_and_scatter_op.cc | 44 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector( in Compile()
46 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("window_strides", in Compile()
|
| D | transpose_op.cc | 57 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("perm", &perm)); in Compile()
164 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(0, &perm)); in InvertPermutation()
|
| D | pooling_ops.cc | 100 auto status = ctx->ConstantInputAsIntVector(1, &ksize); in GetKernelSize()
124 auto status = ctx->ConstantInputAsIntVector(2, &stride); in GetStride()
335 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_)); in Compile()
342 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(4, &stride_)); in Compile()
557 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_)); in Compile()
564 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(4, &stride_)); in Compile()
|
| D | concat_op.cc | 159 ctx->ConstantInputAsIntVector( in Compile()
180 ctx, ctx->ConstantInputAsIntVector( in Compile()
|
| D | empty_op.cc | 50 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("shape", &shape)); in Compile()
|
| D | slice_op.cc | 63 ctx->ConstantInputAsIntVector(1, &begin).ok(); in Compile()
65 ctx->ConstantInputAsIntVector(2, &size).ok(); in Compile()
|
| D | fill_op.cc | 50 ctx->ConstantInputAsIntVector( in Compile()
|
| D | gather_scatter_ops.cc | 40 ctx->ConstantInputAsIntVector("slice_sizes", &slice_sizes)); in Compile()
|
| D | tile_ops.cc | 68 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector( in Compile()
|
| D | xla_broadcast_helper_op.cc | 46 OP_REQUIRES_OK(context, context->ConstantInputAsIntVector("broadcast_dims", in Compile()
|
| D | reverse_op.cc | 93 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &axes)); in Compile()
|
| D | dynamic_slice_ops.cc | 110 ctx, ctx->ConstantInputAsIntVector("size_indices", &size_indices)); in Compile()
|
| D | tensor_array_ops.cc | 316 Status status = ctx->ConstantInputAsIntVector(1, &const_indices); in Compile()
387 Status status = ctx->ConstantInputAsIntVector(1, &const_indices); in Compile()
490 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(2, &lengths)); in Compile()
|
| D | fft_ops.cc | 56 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &fft_length)); in Compile()
|
| D | reshape_op.cc | 50 ctx->ConstantInputAsIntVector( in Compile()
|
| D | spacetobatch_op.cc | 154 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &block_shape)); in Compile()
|
| D | split_op.cc | 147 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &split_sizes)); in Compile()
|
| D | batchtospace_op.cc | 152 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &block_shape)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 171 Status ConstantInputAsIntVector(
174 Status ConstantInputAsIntVector(
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | image_resize_ops.cc | 41 auto status = ctx->ConstantInputAsIntVector(1, &out_size); in GetOutputShape()
|