Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbroadcast_to_op.cc40 context, context->ResolveInputDynamismIntoPredVector(1, &dynamic_dims)); in Compile()
Dreshape_op.cc126 ctx, ctx->ResolveInputDynamismIntoPredVector(1, &dims_are_dynamic)); in Compile()
Dstrided_slice_op.cc241 ctx, ctx->ResolveInputDynamismIntoPredVector(1, &begins_are_dynamic)); in Compile()
244 ctx, ctx->ResolveInputDynamismIntoPredVector(2, &ends_are_dynamic)); in Compile()
Dtensor_list_ops.cc59 ctx->ResolveInputDynamismIntoPredVector(0, &dims_are_dynamic)); in GetTensorListDynamicDims()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h121 Status ResolveInputDynamismIntoPredVector(int index, std::vector<bool>* out);
Dxla_op_kernel.cc291 Status XlaOpKernelContext::ResolveInputDynamismIntoPredVector( in ResolveInputDynamismIntoPredVector() function in tensorflow::XlaOpKernelContext