Searched refs:GetStrides (Results 1 – 3 of 3) sorted by relevance
46 std::pair<INDICES_TYPE, INDICES_TYPE> GetStrides( in GetStrides() function134 GetStrides<INDICES_TYPE>(input_shape, segment_id_shape); in Compute()
34 const gtl::InlinedVector<int64, 8> GetStrides(const TensorShape& shape) { in GetStrides() function152 const auto strides = GetStrides(input_shape); in Compute()
165 std::vector<int64> GetStrides(const OpInfo& op_info) { in GetStrides() function960 std::vector<int64> strides = GetStrides(op_info); in ConvolutionDimensionsFromInputs()2156 std::vector<int64> strides = GetStrides(op_info); in OpDimensionsFromInputs()