Searched refs:element_dims (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | tensor_list_ops.cc | 487 std::vector<int64> element_dims = in Compile() local 490 ctx, element_dims.size() > 1, in Compile() 492 int64 num_elements = element_dims[0]; in Compile() 493 int64 tensor_lengths = element_dims[1]; in Compile() 497 for (int i = 2; i < element_dims.size(); i++) { in Compile() 498 new_dims.push_back(element_dims[i]); in Compile() 533 std::vector<int64> element_dims = in Compile() local 536 ctx, !element_dims.empty(), in Compile() 549 ctx, element_dims[0] % length == 0, in Compile() 551 std::vector<int64> new_dims = {element_dims[0] / length, length}; in Compile() [all …]
|
D | tensor_list_utils.cc | 392 std::vector<int64> element_dims = in ExecuteTensorListPushBack() local 394 element_dims.insert(element_dims.begin(), 1); in ExecuteTensorListPushBack() 395 xla::XlaOp update = xla::Reshape(element, element_dims); in ExecuteTensorListPushBack() 479 std::vector<int64> element_dims = in ExecuteTensorListSetItem() local 481 element_dims.insert(element_dims.begin(), 1); in ExecuteTensorListSetItem() 482 xla::XlaOp update = xla::Reshape(element, element_dims); in ExecuteTensorListSetItem()
|