Searched refs:shape_length (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 96 tensorflow::string ShapeString(const void* shape_ptr, xla::int32 shape_length) { in ShapeString() argument
98 xla::llvm_ir::DecodeSelfDescribingShapeConstant(shape_ptr, shape_length); in ShapeString()
110 const void* shape, xla::int32 shape_length) { in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() argument
115 << ShapeString(shape, shape_length) << " on stream executor " in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
127 << ShapeString(shape, shape_length); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
134 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() argument
139 << ShapeString(shape_ptr, shape_length) << " on stream executor " in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
145 xla::llvm_ir::DecodeSelfDescribingShapeConstant(shape_ptr, shape_length); in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
153 const void* shape_ptr, xla::int32 shape_length) { in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() argument
158 << ShapeString(shape_ptr, shape_length) << " on stream executor " in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
[all …]
|
| D | cpu_runtime.h | 108 const void* shape, xla::int32 shape_length);
124 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length);
130 const void* shape_ptr, xla::int32 shape_length);
143 void* buffer_ptr, const void* shape_ptr, xla::int32 shape_length);
|
| D | ir_emitter.cc | 395 int32 shape_length; in EmitXfeedTransfer() local
398 llvm_ir::EncodeSelfDescribingShapeConstant(shape, &shape_length, &b_)); in EmitXfeedTransfer()
456 shape_ptr, b_.getInt32(shape_length)}); in EmitXfeedTransfer()
469 acquired_pointer, shape_ptr, b_.getInt32(shape_length)}); in EmitXfeedTransfer()
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | rnn_cell.py | 2203 shape_length = len(shapes[0])
2216 if shape_length == 3:
2219 elif shape_length == 4:
2221 strides = shape_length * [1]
2222 elif shape_length == 5:
2224 strides = shape_length * [1]
2233 array_ops.concat(axis=shape_length - 1, values=args),
|