Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable.h61 int64 ShapeSize(const Shape& shape) override { in ShapeSize() function
Dtpu_op_executable.cc118 int64 TpuOpExecutable::ShapeSize(const xla::Shape& shape) { in ShapeSize() function in tensorflow::TpuOpExecutable
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.h220 int64 ShapeSize(const Shape& shape) const { in ShapeSize() function
Dhlo_cost_analysis_test.cc42 int64 ShapeSize(const Shape& shape) { in ShapeSize() function
Dmemory_space_assignment_test.cc43 int64 ShapeSize(const Shape& shape) { in ShapeSize() function
/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
Donedevice_train_palm_detection_filter_inference.cc113 inline int32_t ShapeSize(const int32_t rank, const int32_t* shape) { in ShapeSize() function
Donedevice_train_palm_detection_filter_inference_beta.cc112 inline int32_t ShapeSize(const int32_t rank, const int32_t* shape) { in ShapeSize() function
Donedevice_train_palm_detection_filter_inference_v2.cc112 inline int32_t ShapeSize(const int32_t rank, const int32_t* shape) { in ShapeSize() function