Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DBinaryOp.cpp176 uvec4 input_tensor_size; in binary_op_tensor() member
344 uvec4 input_tensor_size; in binary_op_tensor_() member
/aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/
H A Detnaviv_ml.h61 unsigned input_tensor_size; member
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.cc1588 std::vector<int64_t> input_tensor_size; in CalculateInputTensorSize() local