Home
last modified time | relevance | path

Searched refs:operand_data (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc49 const float* const operand_data = operand_buffer.data.data(); in FuseAddOrSubParamsIntoPrecedingAffine() local
76 const float operand_val = operand_data[operand_channel]; in FuseAddOrSubParamsIntoPrecedingAffine()
120 const float* const operand_data = operand_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
154 bias_data[c] *= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
156 bias_data[c] /= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
165 operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
168 operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
176 weights_data[k * output_depth + c] *= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
178 weights_data[k * output_depth + c] /= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc63 std::vector<const void*> operand_data; in HandleEvaluatorCustomCall() local
64 operand_data.reserve(operands.size()); in HandleEvaluatorCustomCall()
66 operand_data.push_back(literal->untyped_data()); in HandleEvaluatorCustomCall()
73 (*typed_fn)(output_data, operand_data.data()); in HandleEvaluatorCustomCall()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dfusion_test.cc71 Array2D<float> operand_data[Arity]; in TestElementwise2D() local
73 new (&operand_data[i]) Array2D<float>(test_width, test_height); in TestElementwise2D()
81 operand_data[k](i, j) = xs[k]; in TestElementwise2D()
99 LiteralUtil::CreateR2FromArray2D(operand_data[i]))); in TestElementwise2D()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc859 llvm::Value* operand_data = in HandleSelectAndScatter() local
861 Store(operand_data, selected_value_address); in HandleSelectAndScatter()
2123 llvm::Value* operand_data = in HandlePad() local
2143 output_array.EmitWriteArrayElement(output_index, operand_data, &b_); in HandlePad()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc902 llvm::Value* operand_data = in HandleSelectAndScatter() local
904 Store(operand_data, selected_value_address); in HandleSelectAndScatter()