Home
last modified time | relevance | path

Searched defs:shape_arr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Ddlpack.cc230 bool IsValidStrideCompactRowMajorData(int64_t* shape_arr, int64_t* stride_arr, in IsValidStrideCompactRowMajorData()
283 std::vector<int64_t>* shape_arr = &tf_dlm_tensor_ctx->shape; in TFE_HandleToDLPack() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc712 SmallVector<int32_t, 8> shape_arr; in matchAndRewrite() local
Dlegalize_tfl.cc1630 SmallVector<int32_t, 8> shape_arr; in matchAndRewrite() local