Home
last modified time | relevance | path

Searched refs:flat_result (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcustom_gradient.py333 flat_result = nest.flatten(result)
334 flat_result_len = len(flat_result)
355 graphs = {getattr(o, "graph", None) for o in flat_result}
374 input_ops=filtered_input_tensors, output_ops=flat_result)
392 all_tensors = flat_result + args + variables
455 flat_result = nest.flatten(result)
457 flat_result = [gen_array_ops.identity(x) for x in flat_result]
482 tape_lib.record_operation(f.__name__, flat_result, recorded_inputs,
484 flat_result = list(flat_result)
485 return nest.pack_sequence_as(result, flat_result)
Drnn.py331 for r, flat_result in zip(result, flat_results):
333 flat_result.append(r)
336 nest.pack_sequence_as(structure=input_, flat_sequence=flat_result)
337 for input_, flat_result in zip(input_seq, flat_results)
Darray_ops.py5183 flat_result = gather(flat_params, flat_indices)
5184 result = reshape(flat_result, concat([indices_shape, outer_shape], axis=0))
Dcontrol_flow_ops.py2302 flat_result = nest.flatten(original_body_result, expand_composites=True)
2306 _convert_flows_to_tensorarrays(flat_result, exit_vars))
/external/perfetto/src/base/
Dflat_set_unittest.cc101 auto flat_result = flat_set.insert(val); in TEST() local
103 EXPECT_EQ(flat_result.first, flat_set.find(val)); in TEST()
105 EXPECT_EQ(flat_result.second, gold_result.second); in TEST()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_string_ops.py441 flat_result = decode_op(
449 codepoints = flat_result.char_values
451 offsets = flat_result.char_to_byte_starts
454 flat_result.char_values, flat_result.row_splits, validate=False)
459 flat_result.char_to_byte_starts, flat_result.row_splits,
/external/tensorflow/tensorflow/core/kernels/
Dquantization_utils.h742 auto flat_result = result->flat<T>(); in FloatTensorToQuantizedInPlaceUsingEigen() local
743 DCHECK_EQ(flat_input.size(), flat_result.size()); in FloatTensorToQuantizedInPlaceUsingEigen()
746 flat_result.device(device) = QUANTIZE_WITH_EIGEN(flat_input, f2q, T); in FloatTensorToQuantizedInPlaceUsingEigen()
754 auto flat_result = result->flat<T>(); in FloatTensorToQuantizedInPlace() local
756 DCHECK(data_size == flat_result.size()); in FloatTensorToQuantizedInPlace()
758 flat_result(i) = FloatToQuantized<T>(flat_input(i), min, max); in FloatTensorToQuantizedInPlace()
776 auto flat_result = result->flat<float>(); in QuantizedTensorToFloatInPlaceUsingEigen() local
778 DCHECK(data_size == flat_result.size()); in QuantizedTensorToFloatInPlaceUsingEigen()
781 flat_result.device(device) = DEQUANTIZE_WITH_EIGEN(flat_input, q2f); in QuantizedTensorToFloatInPlaceUsingEigen()
790 auto flat_result = result->flat<float>(); in QuantizedTensorToFloatInPlace() local
[all …]
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py570 flat_result = nest.flatten(result)
571 flat_result = [gen_array_ops.identity(x) for x in flat_result]
572 result = nest.pack_sequence_as(result, flat_result)
Dpywrap_tfe_src.cc3875 tensorflow::Safe_PyObjectPtr flat_result(PyList_New(num_retvals)); local
3877 PyList_SET_ITEM(flat_result.get(), i, EagerTensorFromHandle(retvals[i]));
3884 *flattened_inputs, *flattened_attrs, flat_result.get())) {
3897 return flat_result.release();
3899 auto* result = PyList_GET_ITEM(flat_result.get(), 0);
3913 PyObject* obj = PyList_GET_ITEM(flat_result.get(), flat_result_index++);
3922 PyObject* obj = PyList_GET_ITEM(flat_result.get(), flat_result_index++);
3928 PyObject* obj = PyList_GET_ITEM(flat_result.get(), flat_result_index++);
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/
Dbuffer_reuse.mlir361 %flat_result = alloc(%arg_size) : memref<?xi64>
365 } ins(%flat_arg : memref<?xi64>) outs(%flat_result : memref<?xi64>) {
373 %result = memref_reshape %flat_result(%arg_shape)