Searched refs:transposed_tensor (Results 1 – 4 of 4) sorted by relevance
87 Tensor transposed_tensor; in TransposeTensor() local95 TF_RETURN_IF_ERROR(transposed_tensor.BitcastFrom( in TransposeTensor()97 return transposed_tensor; in TransposeTensor()104 &transposed_tensor, alloc_attr)); in TransposeTensor()109 &transposed_tensor)); in TransposeTensor()111 return transposed_tensor; in TransposeTensor()203 Tensor transposed_tensor; in AutoTransposeAndLinearize() local206 TF_ASSIGN_OR_RETURN(transposed_tensor, in AutoTransposeAndLinearize()208 tensor = &transposed_tensor; in AutoTransposeAndLinearize()433 Tensor transposed_tensor; in DoWork() local[all …]
24 float* transposed_tensor) { in Transpose2DTensor() argument25 float* result = transposed_tensor; in Transpose2DTensor()
29 float* transposed_tensor);
229 std::unique_ptr<float[]> transposed_tensor( in MaybeMergeConstNodes() local232 dim0_size, dim1_size, transposed_tensor.get()); in MaybeMergeConstNodes()239 std::string(reinterpret_cast<const char*>(transposed_tensor.get()), in MaybeMergeConstNodes()