Searched defs:new_dimensions (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | canonicalize_all_gather_for_cse.cc | 74 std::vector<int64> new_dimensions(real_data->shape().dimensions().begin(), in RunOnComputation() local
|
| /external/libvpx/libvpx/vp9/encoder/ |
| D | vp9_lookahead.c | 108 int larger_dimensions, new_dimensions; in vp9_lookahead_push() local
|
| /external/libaom/libaom/av1/encoder/ |
| D | lookahead.c | 97 int larger_dimensions, new_dimensions; in av1_lookahead_push() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 352 std::vector<int64> new_dimensions(activations->shape().dimensions().begin(), in HaloDuplicateWithSlice() local
460 std::vector<int64> new_dimensions(activations->shape().dimensions().begin(), in HaloDuplicateWithSlice() local
573 std::vector<int64> new_dimensions(activations->shape().dimensions().begin(), in IncreaseSpatialSizeOnSpaceToBatchedShape() local
1277 std::vector<int64> new_dimensions( in Propagate() local
1731 std::vector<int64> new_dimensions(new_instr->shape().dimensions().begin(), in BatchToSpace() local
|
| D | shape_inference.cc | 432 std::vector<int64> new_dimensions(arg_shape->dimensions().begin(), in InferConcatOpShape() local
2081 std::vector<int64> new_dimensions(shape.dimensions().begin(), in InferAllToAllShape() local
2165 std::vector<int64> new_dimensions; in InferReduceShape() local
|
| D | algebraic_simplifier.cc | 3140 std::vector<int64> new_dimensions; in HandleBroadcast() local
4631 std::vector<int64> new_dimensions; in HandleReduce() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.cc | 345 absl::Span<const int64> new_dimensions, in ReshapeSlice()
|