Home
last modified time | relevance | path

Searched refs:new_dimensions (Results 1 – 9 of 9) sorted by relevance

/external/libvpx/libvpx/vp9/encoder/
Dvp9_lookahead.c108 int larger_dimensions, new_dimensions; in vp9_lookahead_push() local
118 new_dimensions = width != buf->img.y_crop_width || in vp9_lookahead_push()
125 assert(!larger_dimensions || new_dimensions); in vp9_lookahead_push()
135 if (!new_dimensions && ctx->max_sz == 1 && active_map && !flags) { in vp9_lookahead_push()
179 } else if (new_dimensions) { in vp9_lookahead_push()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dcanonicalize_all_gather_for_cse.cc74 std::vector<int64> new_dimensions(real_data->shape().dimensions().begin(), in RunOnComputation() local
76 new_dimensions[0] *= all_gather_participants; in RunOnComputation()
83 new_dimensions), in RunOnComputation()
/external/libaom/libaom/av1/encoder/
Dlookahead.c97 int larger_dimensions, new_dimensions; in av1_lookahead_push() local
108 new_dimensions = width != buf->img.y_crop_width || in av1_lookahead_push()
115 assert(!larger_dimensions || new_dimensions); in av1_lookahead_push()
126 } else if (new_dimensions) { in av1_lookahead_push()
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc352 std::vector<int64> new_dimensions(activations->shape().dimensions().begin(), in HaloDuplicateWithSlice() local
354 new_dimensions[activations_batch_dim] = kNumSplits; in HaloDuplicateWithSlice()
355 new_dimensions.insert(new_dimensions.begin() + activations_batch_dim, in HaloDuplicateWithSlice()
360 MakeReshapeHlo(new_dimensions, activations)); in HaloDuplicateWithSlice()
460 std::vector<int64> new_dimensions(activations->shape().dimensions().begin(), in HaloDuplicateWithSlice() local
462 new_dimensions[activations_batch_dim] = original_batch_size * kNumSplits; in HaloDuplicateWithSlice()
463 new_dimensions.erase(new_dimensions.begin() + activations_batch_dim - 1); in HaloDuplicateWithSlice()
467 MakeReshapeHlo(new_dimensions, activations)); in HaloDuplicateWithSlice()
573 std::vector<int64> new_dimensions(activations->shape().dimensions().begin(), in IncreaseSpatialSizeOnSpaceToBatchedShape() local
583 new_dimensions[spatial_dimension] = reshaped_space_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()
[all …]
Dshape_inference.cc432 std::vector<int64> new_dimensions(arg_shape->dimensions().begin(), in InferConcatOpShape() local
435 new_dimensions[dimension] += arg_shapes[i]->dimensions(dimension); in InferConcatOpShape()
438 Shape result = ShapeUtil::MakeShape(element_type, new_dimensions); in InferConcatOpShape()
2081 std::vector<int64> new_dimensions(shape.dimensions().begin(), in InferAllToAllShape() local
2083 new_dimensions[split_dimension] /= split_count; in InferAllToAllShape()
2084 new_dimensions[concat_dimension] *= split_count; in InferAllToAllShape()
2085 return ShapeUtil::MakeShape(shape.element_type(), new_dimensions); in InferAllToAllShape()
2165 std::vector<int64> new_dimensions; in InferReduceShape() local
2169 new_dimensions.push_back(arg.dimensions(i)); in InferReduceShape()
2176 new_dimensions, new_is_dynamic); in InferReduceShape()
[all …]
Dalgebraic_simplifier.cc3140 std::vector<int64> new_dimensions; in HandleBroadcast() local
3142 new_dimensions.push_back(dims[dim]); in HandleBroadcast()
3147 broadcast->shape(), operand->mutable_operand(0), new_dimensions)); in HandleBroadcast()
4631 std::vector<int64> new_dimensions; in HandleReduce() local
4632 new_dimensions.reserve(arg->dimensions().size() + in HandleReduce()
4635 reduce_dims.end(), std::back_inserter(new_dimensions)); in HandleReduce()
4639 new_dimensions, function)); in HandleReduce()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc345 absl::Span<const int64> new_dimensions, in ReshapeSlice() argument
348 for (int64 i = 0, end = new_dimensions.size(); i < end; ++i) { in ReshapeSlice()
349 new_num_elements *= new_dimensions[i]; in ReshapeSlice()
352 CHECK_EQ(new_dimensions.size(), minor_to_major.size()); in ReshapeSlice()
355 ShapeUtil::MakeShape(literal.shape().element_type(), new_dimensions)); in ReshapeSlice()
Darray.h499 void Reshape(absl::Span<const int64> new_dimensions) {
501 sizes_ = std::vector<int64>(new_dimensions.begin(), new_dimensions.end());
Dliteral_util.h250 static Literal ReshapeSlice(absl::Span<const int64> new_dimensions,