Searched refs:copy_shape (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 159 im2col_array.copy_shape(Shape{output_shape.dims(0), output_shape.dims(1), in ProcessConvOperator()
247 im2col_array.copy_shape( in ProcessTransposeConvOperator()
314 .copy_shape(Shape({batch, height * block_size, width * block_size, in ProcessDepthToSpaceOperator()
338 .copy_shape(Shape({batch, height / block_size, width / block_size, in ProcessSpaceToDepthOperator()
392 output_array.copy_shape(Shape({matmul_repeats, weights_output_depth})); in ProcessFullyConnectedOperator()
478 output_array.copy_shape(input_array.shape()); in ProcessSimpleOperator()
507 output_array.copy_shape(input1_array.shape()); in ProcessSelectOperator()
529 output_array.copy_shape(input0_array.shape()); in ProcessAddNOperator()
603 output_array.copy_shape(input_shape); in ProcessTensorFlowReductionOperator()
605 output_array.copy_shape(Shape({})); in ProcessTensorFlowReductionOperator()
[all …]
|
| D | resolve_batch_normalization.cc | 110 mul_param_array.copy_shape(param_shape); in Run()
111 add_param_array.copy_shape(param_shape); in Run()
|
| D | identify_lstm_merge_inputs.cc | 83 array.copy_shape(shape); in Run()
123 bias_array.copy_shape(Shape({weights_dim1})); in Run()
|
| D | lstm_utils.cc | 47 array.copy_shape(shape); in CreateFloatArrayBuffer()
|
| D | resolve_strided_slice_attributes.cc | 28 attribute_array->copy_shape(strided_slice_shape); in PadAttributeArray()
|
| D | dequantize.cc | 158 new_array.copy_shape(array->shape()); in DequantizeArray()
|
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| D | transform.py | 131 def copy_op_handler(info, op, new_inputs, copy_shape=False, nodedef_fn=None): argument
179 if copy_shape:
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion_test.cc | 139 Shape copy_shape = constant->shape(); in TEST_F() local
140 *copy_shape.mutable_layout() = reversed_layout; in TEST_F()
142 HloInstruction::CreateUnary(copy_shape, HloOpcode::kCopy, constant)); in TEST_F()
144 HloInstruction::CreateUnary(copy_shape, HloOpcode::kCopy, constant)); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | model.h | 2181 void copy_shape(const Shape& src_shape) { *mutable_shape() = src_shape; }
|
| D | tooling_util.cc | 1333 target_array->copy_shape(source_array.shape()); in CopyArrayAttribs()
1399 target_array.copy_shape(source_array.shape()); in CloneArray()
|
| D | export_tensorflow.cc | 2493 wrapped_array.copy_shape(array.shape()); in EncodeConstantArraysMinMaxByWrappingThemInFakeQuantNodes()
|
| /external/python/cpython3/Objects/ |
| D | memoryobject.c | 1239 copy_shape(Py_ssize_t *shape, const PyObject *seq, Py_ssize_t ndim, in copy_shape() function
1296 len = copy_shape(view->shape, shape, ndim, view->itemsize); in cast_to_ND()
|