Home
last modified time | relevance | path

Searched refs:NumMappedDims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.h47 inline constexpr int64_t NumMappedDims() { return 3; } in NumMappedDims() function
Dspace_to_batch_converter.cc1982 std::vector<int64_t> dim_map(NumMappedDims()); in Propagate()
2725 std::vector<int64_t> dim_map(NumMappedDims()); in PropagateOnConv()
3441 std::vector<int64_t> dim_map(NumMappedDims()); in PropagateOnBackpropFilterConv()
3824 std::vector<int64_t> dim_map(NumMappedDims()); in PerformSpaceToBatchOnConvolution()