Searched refs:BuildTensorList (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 113 ctx, BuildTensorList( in Compile()
150 ctx, BuildTensorList(buffer, xla::ConstantR0<int32>(ctx->builder(), 0), in Compile()
296 ctx, BuildTensorList(tensor, xla::ConstantR0<int32>(b, num_elements), in Compile()
371 OP_REQUIRES_OK(ctx, BuildTensorList(xla::DynamicUpdateSlice(buffer, update, in Compile()
416 BuildTensorList(xla::DynamicUpdateSlice(buffer, update, start_indices), in Compile()
461 OP_REQUIRES_OK(ctx, BuildTensorList(ta, index, &output_list)); in Compile()
|
| D | tensor_list_utils.cc | 31 Status BuildTensorList(const xla::XlaOp& buffer, const xla::XlaOp& push_index, in BuildTensorList() function
97 return BuildTensorList(new_buffer, push_index, output_list); in InitializeTensorList()
|
| D | tensor_list_utils.h | 35 Status BuildTensorList(const xla::XlaOp& buffer, const xla::XlaOp& push_index,
|
| D | aggregate_ops.cc | 60 OP_REQUIRES_OK(ctx, BuildTensorList(sum, push_index, &sum)); in Compile()
|