Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/shim/
Dtest_util.h72 TfLiteIntArray* shape_arr = TfLiteIntArrayCreate(shape.size()); in ReallocDynamicTensor() local
/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.cc716 SmallVector<int32_t> shape_arr; in matchAndRewrite() local
Dlegalize_tfl.cc1505 SmallVector<int32_t> shape_arr; in matchAndRewrite() local