Home
last modified time | relevance | path

Searched defs:input_tensor (Results 101 – 125 of 280) sorted by relevance

12345678910>>...12

/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/
H A Dconv_activation.py142 def relu1(input_tensor): argument
H A Dconv_bias_activation.py65 def add_conv(input_tensor, filter_input, parameters): argument
H A Dpool.py115 def make_l2_pool(input_tensor, ksize, strides, padding, data_format): argument
/aosp_15_r20/external/federated-compute/fcp/artifact_building/
H A Dplan_utils.py65 def create_input(input_tensor): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_util_platform_specific.cc42 void GPUDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice()
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DSplitLayerSubTensorMutator.cpp68 Tensor *input_tensor = node->input(0); in mutate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
H A Dimporter_test_min_max.cc123 auto& input_tensor = sub_graph->tensors[op->inputs[0]]; in InjectStatsToFullyConnected() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_context.cc43 void PluggableDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/fuzzing/
H A Done_hot_fuzz.cc63 Tensor input_tensor(tensorflow::DT_UINT8, TensorShape({input_size})); in FuzzImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/
H A Dconv_bias_activation.py69 def add_conv(input_tensor, filter_input, parameters): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbroadcast_to_op.cc44 const Tensor& input_tensor = ctx->input(0); in Compute() local
H A Dbucketize_op.cc63 const Tensor& input_tensor = context->input(0); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/examples/v1/
H A Ddebug_mnist_v1.py158 def nn_layer(input_tensor, input_dim, output_dim, layer_name, act=tf.nn.relu): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/signal/
H A Ddct_ops.py28 def _validate_dct_arguments(input_tensor, dct_type, n, axis, norm): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Darg_min_max_builder.cc35 const auto& input_tensor = context->tensors[input_tensor_id]; in PopulateSubGraph() local
H A Dmirror_pad_builder.cc34 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
H A Dsplit_builder.cc32 const auto& input_tensor = context->tensors[input_tensor_id]; in PopulateSubGraph() local
H A Dresize_nearest_neighbor_builder.cc33 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_tfconv_op.h69 const Tensor& input_tensor = MklGetInput(context, input_number); in ConvertMklToTf() local
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/examples/v2/
H A Ddebug_mnist_v2.py170 def dense_layer(weights, input_tensor, act=tf.nn.relu): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dsingle_threaded_cpu_device.cc72 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice()
/aosp_15_r20/external/tensorflow/tensorflow/examples/speech_commands/
H A Dfreeze.py172 def save_saved_model(file_name, sess, input_tensor, output_tensor): argument
/aosp_15_r20/external/libtextclassifier/native/utils/
H A Dtflite-model-executor.h95 TfLiteTensor* input_tensor = in SetInput() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DUnstackFixture.h66 TensorType input_tensor = create_tensor<TensorType>(input_shape, data_type); in compute_target() local
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dinput_layer.py102 input_tensor=None, argument

12345678910>>...12