Searched refs:DynamicShapeIsCompatible (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | if_op.cc | 119 if (!xla::ShapeUtil::DynamicShapeIsCompatible(then_result.xla_output_shape, in ValidateShapes()
121 !xla::ShapeUtil::DynamicShapeIsCompatible(else_result.xla_output_shape, in ValidateShapes()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 169 bool DynamicShapeIsCompatible(const xla::Shape& dynamic_shape, in DynamicShapeIsCompatible() function
225 DynamicShapeIsCompatible(runtime_shape, compile_time_shape)); in UpdateDynamicInputs()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_transfer_manager.cc | 185 TF_RET_CHECK(ShapeUtil::DynamicShapeIsCompatible(*device_shape, in ReadDynamicShapes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_xfeed.cc | 313 TF_RET_CHECK(ShapeUtil::DynamicShapeIsCompatible(*device_shape, in ReadDynamicShapesOnCpu()
|
| D | cpu_executable.cc | 335 ShapeUtil::DynamicShapeIsCompatible(actual_shape, expected_shape)) in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 46 if (!ShapeUtil::DynamicShapeIsCompatible(input_shape, dynamic_shape)) { in SetDynamicShape()
|
| D | transfer_manager.cc | 246 TF_RET_CHECK(ShapeUtil::DynamicShapeIsCompatible(*device_shape, in ReadDynamicShapes()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 614 static bool DynamicShapeIsCompatible(const xla::Shape& dynamic_shape,
|
| D | shape_util.cc | 1608 /* static */ bool ShapeUtil::DynamicShapeIsCompatible( in DynamicShapeIsCompatible() function in xla::ShapeUtil
|
| D | literal.cc | 701 CHECK(ShapeUtil::DynamicShapeIsCompatible(compact_shape, bound_shape)) in CopyFrom()
|