Home
last modified time | relevance | path

Searched refs:ResolveInputDynamismIntoPredVector (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_op_kernel.h130 Status ResolveInputDynamismIntoPredVector(int index, std::vector<bool>* out);
132 Status ResolveInputDynamismIntoPredVector(absl::string_view name,
H A Dxla_op_kernel.cc300 Status XlaOpKernelContext::ResolveInputDynamismIntoPredVector( in ResolveInputDynamismIntoPredVector() function in tensorflow::XlaOpKernelContext
303 return ResolveInputDynamismIntoPredVector(index, out); in ResolveInputDynamismIntoPredVector()
345 Status XlaOpKernelContext::ResolveInputDynamismIntoPredVector( in ResolveInputDynamismIntoPredVector() function in tensorflow::XlaOpKernelContext
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbroadcast_to_op.cc42 context, context->ResolveInputDynamismIntoPredVector(1, &dynamic_dims)); in Compile()
H A Dfill_op.cc54 ctx, ctx->ResolveInputDynamismIntoPredVector("dims", &dynamic_dims)); in Compile()
H A Dstrided_slice_op.cc277 ctx, ctx->ResolveInputDynamismIntoPredVector(1, &begins_are_dynamic)); in Compile()
280 ctx, ctx->ResolveInputDynamismIntoPredVector(2, &ends_are_dynamic)); in Compile()
608 ctx->ResolveInputDynamismIntoPredVector(0, &dynamic_input)); in Compile()
H A Dtile_ops.cc82 OP_REQUIRES_OK(ctx, ctx->ResolveInputDynamismIntoPredVector( in Compile()
H A Dsparse_to_dense_op.cc79 context, context->ResolveInputDynamismIntoPredVector(1, &dynamic_dims)); in Compile()
H A Dreshape_op.cc144 ctx, ctx->ResolveInputDynamismIntoPredVector(1, &dims_are_dynamic)); in Compile()
H A Dslice_op.cc109 ctx, ctx->ResolveInputDynamismIntoPredVector(2, &size_is_dynamic)); in Compile()
H A Dtensor_list_ops.cc60 ctx->ResolveInputDynamismIntoPredVector(0, &dims_are_dynamic)); in GetTensorListDynamicDims()