Searched defs:local_shape (Results 1 – 8 of 8) sorted by relevance
321 const std::vector<int64_t> local_shape() const { return local_shape_; } in local_shape() function430 std::vector<int64_t> local_shape) in ResourceHandleWithLayout()460 const std::vector<int64_t>& local_shape, in Dummy()506 : TensorWithLayout(nullptr, mesh, layout, local_shape), in TensorWithLayout() argument
91 std::vector<int64_t> local_shape; in Compute() local
347 const std::vector<int64_t>& local_shape, const TF_DataType dtype, in Dummy()438 std::vector<int64_t> local_shape) { in Wrap()
991 std::vector<int64_t> local_shape; in LocalShapeFromGlobalShape() local1006 PartialTensorShape local_shape({}); in LocalShapeFromGlobalShape() local
975 std::vector<int64_t> local_shape(data, data + shape_tensor_size); in SparsePack() local1718 std::vector<int64_t> local_shape = in ExecuteRegularOperation() local
31 auto local_shape = computed_layout->LocalShapeFromGlobalShape( in ExpandOp() local
34 std::vector<int64_t> local_shape = in ExpandOp() local
297 auto local_shape = output_layout.LocalShapeFromGlobalShape(global_shape); in HandleConvBackpropInput() local