Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable.cc118 int64 TpuExecutable::ShapeSize(const Shape& shape) { in ShapeSize() function in xla::TpuExecutable
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.h215 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.cc34 int64 ShapeSize(const Shape& shape) { in ShapeSize() function
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc202 int64 ShapeSize(const Shape& shape) override { in ShapeSize() function in xla::__anoncef202870111::TpuExecutable