Home
last modified time | relevance | path

Searched refs:ConstantInputReshapedToIntVector (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dcross_replica_ops.cc57 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector( in Compile()
78 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector( in Compile()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_op_kernel.h180 Status ConstantInputReshapedToIntVector(
183 Status ConstantInputReshapedToIntVector(
H A Dxla_op_kernel.cc390 Status XlaOpKernelContext::ConstantInputReshapedToIntVector( in ConstantInputReshapedToIntVector() function in tensorflow::XlaOpKernelContext
398 Status XlaOpKernelContext::ConstantInputReshapedToIntVector( in ConstantInputReshapedToIntVector() function in tensorflow::XlaOpKernelContext
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dreduction_ops_common.cc68 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector(1, &axes)); in Compile()
H A Ddynamic_partition_op.cc117 ctx->ConstantInputReshapedToIntVector(1, &partitions_static).ok(); in Compile()
H A Dshape_op.cc265 OP_REQUIRES_OK(ctx, ctx->ConstantInputReshapedToIntVector("dim", &dims)); in Compile()