Searched refs:new_tensor_shape (Results 1 – 3 of 3) sorted by relevance
334 Arrangement new_tensor_shape; in slice_shape() local335 if (new_tensor_shape.Init(shape) == Status::FAILED) { in slice_shape()339 return new_tensor_shape; in slice_shape()
228 Arrangement new_tensor_shape = tensor_layout.slice_shape(); in TEST_F() local231 ASSERT_EQ(new_tensor_shape, expected_shape); in TEST_F()
553 std::vector<int64_t> new_tensor_shape(input_shapes.begin(), input_shapes.end()); in SyncEmbeddingTables() local555 …sor::TensorPtr new_tensor = std::make_shared<tensor::Tensor>(kNumberTypeFloat32, new_tensor_shape); in SyncEmbeddingTables()