Home
last modified time | relevance | path

Searched refs:DynamicShapeIsCompatible (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc167 bool DynamicShapeIsCompatible(const xla::Shape& dynamic_shape, in DynamicShapeIsCompatible() function
223 DynamicShapeIsCompatible(runtime_shape, compile_time_shape)); in UpdateDynamicInputs()
/external/tensorflow/tensorflow/compiler/xla/service/
Dexecutable.cc47 if (!ShapeUtil::DynamicShapeIsCompatible(input_shape, dynamic_shape)) { in SetDynamicShape()
Dtransfer_manager.cc250 TF_RET_CHECK(ShapeUtil::DynamicShapeIsCompatible(*device_shape, in ReadDynamicShapes()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc342 ShapeUtil::DynamicShapeIsCompatible(actual_shape, expected_shape)) in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h701 static bool DynamicShapeIsCompatible(const xla::Shape& dynamic_shape,
Dshape_util.cc1588 /* static */ bool ShapeUtil::DynamicShapeIsCompatible( in DynamicShapeIsCompatible() function in xla::ShapeUtil
Dliteral.cc557 CHECK(ShapeUtil::DynamicShapeIsCompatible(compact_shape, bound_shape)) in CopyFrom()