Searched refs:GetStrides (Results 1 – 7 of 7) sorted by relevance
34 const gtl::InlinedVector<int64, 8> GetStrides(const TensorShape& shape) { in GetStrides() function152 const auto strides = GetStrides(input_shape); in Compute()
195 mkl_shape->GetStrides()); in Compute()207 mkl_shape->GetSizes(), mkl_shape->GetStrides()); in Compute()
222 input_shape.GetStrides()); in MklAllocateOutputs()240 input_shape.GetStrides()); in MklAllocateOutputs()406 tmp_mkl_shape->GetStrides()[i]; in Compute()
97 mkl_context.in_strides[i] = mkl_context.input_shape.GetStrides()[i]; in Compute()307 mkl_context.in_strides[i] = tmp_mkl_shape->GetStrides()[i]; in Compute()
115 mkl_context.in_strides[i] = tmp_mkl_shape->GetStrides()[i]; in Compute()
70 std::vector<int64> GetStrides(const OpInfo& op_features) { in GetStrides() function487 std::vector<int64> strides = GetStrides(op_features); in ConvolutionDimensionsFromInputs()
162 const size_t* GetStrides() const { return strides_; } in GetStrides() function