Home
last modified time | relevance | path

Searched defs:ShapeSize (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_hlo_cost_analysis_test.cc25 int64_t ShapeSize(const Shape& shape) { in ShapeSize() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executable.h61 int64_t ShapeSize(const Shape& shape) override { in ShapeSize() function
H A Dtpu_op_executable.cc118 int64_t TpuOpExecutable::ShapeSize(const xla::Shape& shape) { in ShapeSize() function in tensorflow::TpuOpExecutable
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/tpu/
H A Dtpu_op_executable.cc118 int64 TpuOpExecutable::ShapeSize(const xla::Shape& shape) { in ShapeSize() function in tensorflow::TpuOpExecutable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_verifier.h97 int64_t ShapeSize(const Shape& shape) const { return shape_size(shape); } in ShapeSize() function
H A Dhlo_cost_analysis_test.cc42 int64_t ShapeSize(const Shape& shape) { in ShapeSize() function
H A Dmemory_space_assignment_test.cc44 int64_t ShapeSize(const Shape& shape) { in ShapeSize() function
/aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
H A Donedevice_train_palm_detection_filter_inference_beta.cc112 inline int32_t ShapeSize(const int32_t rank, const int32_t* shape) { in ShapeSize() function
H A Donedevice_train_palm_detection_filter_inference.cc113 inline int32_t ShapeSize(const int32_t rank, const int32_t* shape) { in ShapeSize() function
H A Donedevice_train_palm_detection_filter_inference_v2.cc112 inline int32_t ShapeSize(const int32_t rank, const int32_t* shape) { in ShapeSize() function