Home
last modified time | relevance | path

Searched refs:element_dims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc487 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 …]
Dtensor_list_utils.cc392 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()